最近在做一个小项目,多标签图像分类,二分类交叉熵损失,标签是0-1多热向量。模型用的ResNet50,改了最后一层输出为对应类别数。训练时loss一直在0.7左右徘徊,下不去,但验证集上有些标签的AUC还挺高,有的又很低。我怀疑是数据不平衡问题,有些标签出现频率极低,试了给损失加权重,但效果不明显。也试过调整学习率、换优化器,都没啥改善。有没有老哥指点一下,这种情况一般从哪排查?是模型结构问题还是数据问题?或者有更合适的损失函数推荐吗?先谢谢了。