arXiv提出可解释的结构化温度缩放方法，提升图神经网络与点云分割校准性能

摘要：arXiv发布了一种名为可解释结构化温度缩放（ISTS）的新方法，用于提升图神经网络（GNN）和点云分割的校准性能。ISTS通过一个包含12个参数的线性模型，根据节点置信度、邻域和与标注节点的距离等特征设置每个节点温度，从而实现更精细的温度调整。该方法在多个基准测试中表现优异，在负对数似然（NLL）和Brier评分上均优于现有方法，同时显著降低了计算成本并提高了可解释性。

---

技术机制剖析

ISTS通过引入一个包含12个参数的线性模型，根据每个节点的置信度、邻域结构和与标注节点的距离等特征来设置温度。这种方法的核心在于其可解释性：每个温度值都可以追溯到具体特征的贡献，从而提供更细粒度的校准。具体来说，ISTS通过以下步骤实现：

1. 特征提取：从节点及其邻域中提取置信度、邻域结构和距离等特征。
2. 线性模型应用：使用线性模型将这些特征组合成一个温度值。
3. 交叉验证正则化：通过交叉验证确定模型的正则化参数，以确保模型不会过度偏离传统的温度缩放方法。

工程权衡与实测表现

ISTS在多个基准测试中表现出色，包括在负对数似然（NLL）和Brier评分上的优异表现。与传统的温度缩放方法相比，ISTS在NLL上降低了1.8%，这一改进是同等拟合的基于熵的温度方法的2倍以上。此外，ISTS在处理点云分割任务时，显著提升了NLL、Brier评分以及对拒绝点的排序性能。

ISTS的优势在于其计算效率高且可解释性强。与需要辅助网络的复杂校准方法相比，ISTS仅需少量参数即可实现高效校准，且其解释性在节点组级别上具有高保真度。

开发者落地与部署建议

对于开发者而言，ISTS是一种低成本且高效的校准选择，尤其适用于资源受限的应用场景。以下是一些部署建议：

• 资源优化：ISTS的低参数数量使其在计算和内存资源有限的环境中表现出色。
• 可解释性需求：如果需要解释校准结果，ISTS提供了清晰的特征贡献路径。
• 多场景适用性：ISTS不仅适用于图神经网络，还适用于点云分割等需要精细校准的场景。

结论

ISTS是一种创新的校准方法，通过可解释的结构化温度缩放提升了GNN和点云分割的性能。其在多个基准测试中的优异表现和低计算成本使其成为AI模型校准领域的有力工具。

---

消息来源：ArXiv Machine Learning (cs.LG) (2026-11-03)

—— 完 ——

主题标签： #图神经网络 关注 #温度缩放 关注 #校准 关注 #点云分割 关注 #可解释性 关注