Javascript Adam优化器每次只返回一个类

我目前正在使用Tensorflow.js创建一个用于图像分类的web应用程序。我意识到，当我尝试拟合/训练我的模型时，SGD优化器在训练集上进行预测时做得很好。另一方面，Adam优化器训练的模型默认仅为一个类/类别，即使oneHot编码标签中有7个不同的类

我已经尝试过多次降低学习率，但结果仍然相同。有趣的是，尽管在训练集上的预测保持不变，但损失仍在继续下降。为什么在这种情况下只有SGD优化器工作

下面的代码是在按下“Train Model”（列车模型）按钮时执行的异步函数中编写的。产生这两个结果的代码中唯一的区别是使用了优化器（其他一切都是一样的）

结果

Actual Labels:
[[1, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 1, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 0, 0, 1],
[0, 0, 0, 1, 0, 0, 0],
[0, 0, 0, 1, 0, 0, 0]]

SGD Prediction (good accuracy):
[[0.7766488, 0.0011008, 0   , 0.1785417, 0.0128197, 0.0004938, 0.0303952],
     [0.0000667, 0.0000794, 0   , 0.0006806, 0.9991421, 0.0000079, 0.0000238],
     [0.0014632, 0.5784221, 0   , 0.1196582, 0.2930434, 0.0010978, 0.0063155],
     [0.0000067, 0.000024 , 0   , 0.0015023, 0.9984038, 0.0000043, 0.0000589],
     [0.0000328, 0.0000816, 0   , 0.0027291, 0.9970328, 0.0000105, 0.0001136],
     [0.0000657, 0.0000679, 0   , 0.0176743, 0.002068 , 0.000019 , 0.9801047],
     [0.0055083, 0.0023474, 1e-7, 0.7483685, 0.215641 , 0.0009718, 0.027163 ],
     [0.0045981, 0.0015817, 0   , 0.922437 , 0.026959 , 0.0002184, 0.044206 ]]

Adam Prediction (at every step since step 1):
[[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0]]

---

请详细解释您的问题和网络。可能SGD是更好的训练方法，当然也可能您没有使用合适的参数值来训练Adam网络。首先，试着降低学习率

---

祝你好运

如果方便的话，请发布您模型的代码。您能否将完整的模型添加到问题中，以及您设置adam优化器的方式？

Actual Labels:
[[1, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 1, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 0, 0, 1],
[0, 0, 0, 1, 0, 0, 0],
[0, 0, 0, 1, 0, 0, 0]]

SGD Prediction (good accuracy):
[[0.7766488, 0.0011008, 0   , 0.1785417, 0.0128197, 0.0004938, 0.0303952],
     [0.0000667, 0.0000794, 0   , 0.0006806, 0.9991421, 0.0000079, 0.0000238],
     [0.0014632, 0.5784221, 0   , 0.1196582, 0.2930434, 0.0010978, 0.0063155],
     [0.0000067, 0.000024 , 0   , 0.0015023, 0.9984038, 0.0000043, 0.0000589],
     [0.0000328, 0.0000816, 0   , 0.0027291, 0.9970328, 0.0000105, 0.0001136],
     [0.0000657, 0.0000679, 0   , 0.0176743, 0.002068 , 0.000019 , 0.9801047],
     [0.0055083, 0.0023474, 1e-7, 0.7483685, 0.215641 , 0.0009718, 0.027163 ],
     [0.0045981, 0.0015817, 0   , 0.922437 , 0.026959 , 0.0002184, 0.044206 ]]

Adam Prediction (at every step since step 1):
[[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0, 0]]