跳到正文
原文
arXiv· arXiv:2502.01628· David D. Baek, Ziming Liu, Riya Tyagi, Max Tegmark·本站收录· 原文发表

调和损失训练出可解释的 AI 模型

Harmonic Loss Trains Interpretable AI Models

AI 导读

论文提出调和损失(harmonic loss)作为训练神经网络和大语言模型的替代监督信号,其与标准交叉熵的区别在于用尺度不变的 HarMax 函数替代 SoftMax 归一化,并以欧氏距离而非点积计算 logits。作者称,尺度不变性和设计上有限的收敛点(可解释为类中心)带来更好的可解释性与更快的收敛。作者在算法、视觉与语言数据集上验证,称调和损失训练的模型比标准模型可解释性更强、泛化所需数据更少、grokking 减少。与标准 GPT-2 相比,用调和损失训练的 GPT-2 学到更可解释的表示。作者认为,该损失在数据有限或高风险场景中可能有价值。

阅读原文arxiv.org