让模型训练更贴近真实决策,提升边缘检测精度
Binarization-Aware Adjuster for Discrete Decision Learning with an Application to Edge Detection
- 引入可微的二值化感知调节器,让优化过程考虑决策临界区
- 在多个边缘检测数据集上显著提升模型性能
- 适合需要精确离散决策的视觉任务研究者
机器学习中的离散决策任务存在训练与推理不一致的问题:模型以连续输出进行优化,却以离散预测评估。这种不一致源于二值化操作的不连续性,导致决策行为无法直接融入梯度优化。为此,我们提出理论严谨的二值化感知调节器(BAA)框架,将二值化特性嵌入连续优化过程。该框架基于距离加权函数(DWF),根据预测正确性与决策阈值的距离动态调节损失贡献,使优化重点聚焦于决策敏感区域,同时兼容标准学习流程。我们将BAA应用于边缘检测(ED)这一典型二值决策任务,在多个代表性模型和数据集上的实验表明,引入BAA能持续提升性能,验证了其有效性。本工作建立了一种将连续优化与离散决策行为对齐的系统方法,并在具体应用中得到证实。
原文摘要 · Abstract (English)
Discrete decision tasks in machine learning exhibit a fundamental misalignment between training and inference: models are optimized with continuous-valued outputs but evaluated using discrete predictions. This misalignment arises from the discontinuity of discretization operations, which prevents decision behavior from being directly incorporated into gradient-based optimization. To address this issue, we propose a theoretically grounded framework termed the Binarization-Aware Adjuster (BAA), which embeds binarization characteristics into continuous optimization. The framework is built upon the Distance Weight Function (DWF), which modulates loss contributions according to prediction correctness and proximity to the decision threshold, thereby aligning optimization emphasis with decision-critical regions while remaining compatible with standard learning pipelines. We apply the proposed BAA framework to the edge detection (ED) task, a representative binary decision problem. Experimental results on representative models and datasets show that incorporating BAA into optimization leads to consistent performance improvements, supporting its effectiveness. Overall, this work establishes a principled approach for aligning continuous optimization with discrete decision behavior, with its effectiveness demonstrated in a concrete application setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。