提升Transformer检测线段精度,解决低置信度抑制问题。
Improving Transformer Based Line Segment Detection with Matched Predicting and Re-ranking
- 用可学习几何信息后验优化预测排名,提高高质量线段置信度。
- 新提案方法使特征点直接预测线段位置,训练更快更稳。
- 引入排序损失增强泛化能力,训练轮数少于同类模型。
基于Transformer的线段检测方法虽表现优异,但存在部分高精度线段因置信度低被错误降序甚至抑制的问题。此外,模型常需长时间训练,主要源于二部匹配需求。本文提出RANK-LETR,通过可学习几何信息在后验阶段提升优质预测的置信度,优化线段排序。我们设计新提案策略:以线段质心最近的特征点直接预测位置,显著提升训练效率与稳定性。同时引入线段排序损失,稳定训练过程中的排序行为,增强模型泛化能力。实验表明,该方法在预测精度上优于其他基于Transformer和CNN的方法,且所需训练轮数少于先前的Transformer模型。
原文摘要 · Abstract (English)
Classical Transformer-based line segment detection methods have delivered impressive results. However, we observe that some accurately detected line segments are assigned low confidence scores during prediction, causing them to be ranked lower and potentially suppressed. Additionally, these models often require prolonged training periods to achieve strong performance, largely due to the necessity of bipartite matching. In this paper, we introduce RANK-LETR, a novel Transformer-based line segment detection method. Our approach leverages learnable geometric information to refine the ranking of predicted line segments by enhancing the confidence scores of high-quality predictions in a posterior verification step. We also propose a new line segment proposal method, wherein the feature point nearest to the centroid of the line segment directly predicts the location, significantly improving training efficiency and stability. Moreover, we introduce a line segment ranking loss to stabilize rankings during training, thereby enhancing the generalization capability of the model. Experimental results demonstrate that our method outperforms other Transformer-based and CNN-based approaches in prediction accuracy while requiring fewer training epochs than previous Transformer-based models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。