Apr, 2024
图像分类的下一代损失函数
Next Generation Loss Function for Image Classification
TL;DR神经网络通过最小化损失函数来学习,定义了预测模型输出与目标值之间的差异。选择损失函数对于实现特定任务行为至关重要,并且极大地影响模型的能力。我们利用遗传编程方法对众所周知的损失函数进行了实验性挑战,包括交叉熵损失,提出了5个最佳函数,并在各种模型架构的标准数据集上进行了评估,结果发现Next Generation Loss(NGL)函数在所有测试数据集上表现出相同或更好的性能,包括Imagetnet-1k数据集。最后,我们将NGL函数用于Pascal VOC 2012和COCO-Stuff164k数据集的分割后任务训练,提高了底层模型的性能。