arXiv:2605.20369cs.CLcs.AI2026-05

提出新损失函数,让大模型更准地预测数字。

DEL: Digit Entropy Loss for Numerical Learning of Large Language Models

论文配图:DEL: Digit Entropy Loss for Numerical Learning of Large Language Models
图 1 · 摘自论文原文
  • 用数字条件概率和二元交叉熵实现监督式熵优化
  • 去掉数值距离项,避免分布过尖或过平
  • 支持浮点数预测,适合数学与代码任务

数字预测是大语言模型在数学推理和代码生成中的基础能力。现有基于最大似然估计的训练方式不适用于数字预测。近期的惩罚驱动方法(如Number Token Loss、Discretized Distance Loss)引入数值距离先验,但分别导致分布过尖和过平。本文深入分析大模型数字学习机制,发现现有方法均遵循准则-距离范式:准则项决定优化模式,距离项引入几何先验。为此,我们提出数字熵损失(DEL),通过三项关键设计重构传统无监督熵优化:利用数字条件概率与二元交叉熵将熵优化转为有监督形式;去除距离项以规避数值距离问题;将整数学习泛化至浮点数优化,提升预测精度。DEL可统一处理整数、小数和小数点,将学习目标从单个数字扩展到浮点数域。在包含CodeLlama、Mistral、DeepSeek、Qwen-2.5在内的四款主流大模型上,于七个数学推理基准测试中,DEL在整体预测准确率与数值距离上均持续优于基线方法。源码见https://github.com/PolyU-VCLab/DEL。

原文摘要 · Abstract (English)

Number prediction stands as a fundamental capability of large language models (LLMs) in mathematical problem-solving and code generation. The widely adopted maximum likelihood estimation (MLE) for LLM training is not tailored to number prediction. Recently, penalty-driven approaches, e.g., Number Token Loss and Discretized Distance Loss, introduce an inductive bias of numerical distance but induce over-sharpened and over-flattened digit distributions, respectively. In this paper, we make an in-depth analysis on LLM numerical learning, and show that existing numerical learning methods conceptually follow a criterion-distance formulation, where the criterion term represents optimization pattern and the distance term instills geometric prior. Consequently, we present Digit Entropy Loss (DEL) for auto-regressive numerical learning, which reformulates the conventional unsupervised entropy optimization in three key designs: leveraging digit conditional probability and binary cross-entropy to guide the entropy optimization into a supervised manner; deprecating the distance term to bypass the issue of numerical distance; and generalizing the integer-based numerical learning to floating-point number optimization, enabling more accurate number prediction. Our DEL formulation can incorporate integers, decimals, and decimal points, expanding the learning objective from a single digit to the floating-point number domain. Experiments conducted on seven mathematical reasoning benchmarks with four representative LLMs, including CodeLlama, Mistral, DeepSeek, and Qwen-2.5, demonstrate that DEL consistently outperforms its counterparts in both overall prediction accuracy and numerical distance. Source codes are at https://github.com/PolyU-VCLab/DEL

大模型数字预测损失函数数学推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。