arXiv:2608.30603cs.CVcs.AI2026-08

用扩散模型筛选高质量最小样本集,提速上百倍。

DiffSAC: Diffusion-guided Sampling for Consensus-based Robust Estimation

论文配图:DiffSAC: Diffusion-guided Sampling for Consensus-based Robust Estimation
图 1 · 摘自论文原文
  • 用扩散模型学习有效最小样本分布,评估数据点归属优质集的置信度。
  • 仅需数十个候选集即达顶尖性能,相比前人超万次评估效率大增。
  • 可即插即用,适合需要高效鲁棒估计的视觉任务场景。

鲁棒估计是计算机视觉核心任务,常通过样本共识方法解决。然而传统方法因难以在假设评估前识别有效最小集,导致采样效率低下。为此,我们提出一种基于扩散模型的共识鲁棒估计框架 DiffSAC。DiffSAC 利用扩散模型学习有效最小集的分布,不进行数据点排序,而是直接评估每个点属于优质最小集的置信度,大幅减少无效集合处理。为约束优化方向,几何特征作为条件嵌入扩散模型中。由此,DiffSAC 输出少量高质量最小集,经共识评估即可确定最优假设。显著优于此前需评估逾万次假设的方法,仅需数十次即可实现顶尖性能。在五个经典视觉任务上广泛实验验证其优势。扩散采样加速器支持实时运行,且可作为模块化组件提升现有样本共识方法。

原文摘要 · Abstract (English)

Robust estimation is a core computer vision task frequently tackled using sample consensus. However, traditional methods suffer from inefficient sampling as they struggle to identify effective minimum sets before hypothesis evaluation. To address these challenges, we propose a novel Diffusion-guided Sampling for Consensus-based Robust Estimation (DiffSAC) framework. DiffSAC introduces a diffusion model to learn the distribution of effective minimum sets. It refines the confidence for each data point, indicating whether it belongs to a good minimum set, rather than ranking the data points as in previous work. This significantly reduces the need to process numerous bad sets. To constrain the refinement direction, geometric features are incorporated as conditions within our diffusion model. Consequently, DiffSAC outputs a small number of high-quality minimum sets, enabling identification of the best hypothesis via consensus evaluation. Notably, compared to previous works requiring evaluating over ten thousand hypotheses, DiffSAC achieves state-of-the-art performance with only dozens, significantly boosting efficiency. Extensive experiments across five classic computer vision tasks demonstrate the superiority of DiffSAC. The diffusion model's sampling accelerators enable real-time operation, and DiffSAC can be used as a plug-and-play module to improve existing sample consensus methods.

鲁棒估计扩散模型样本共识

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