arXiv:2602.15405cs.LG2026-02

用两个相互引导的扩散模型,同时提升信号质量和分类准确率。

Joint Enhancement and Classification using Coupled Diffusion Models of Signals and Logits

  • 用信号和输出概率共同建模,双向优化信号与分类结果。
  • 在多种噪声下,图像和语音分类准确率显著高于传统方法。
  • 无需重训练分类器,适合快速部署到现有模型中。

在噪声环境中的鲁棒分类仍是机器学习的核心挑战。传统方法将信号增强与分类分作两个独立步骤:先增强信号,再进行分类。这种方法无法利用分类器输出的语义信息来指导去噪。本文提出一种通用、领域无关的框架,整合两个相互作用的扩散模型:一个作用于输入信号,另一个作用于分类器的输出logits,且无需对分类器进行重新训练或微调。这种耦合机制实现双向引导:增强信号优化类别估计,而不断演化的类别logits则引导信号重建向流形中具有判别性的区域。我们提出了三种有效建模输入与logits联合分布的策略。在图像分类和自动语音识别任务上评估了该联合增强方法。结果表明,所提框架优于传统的顺序增强基线,在多种噪声条件下均显著提升了分类准确率,展现了更强的鲁棒性与灵活性。

原文摘要 · Abstract (English)

Robust classification in noisy environments remains a fundamental challenge in machine learning. Standard approaches typically treat signal enhancement and classification as separate, sequential stages: first enhancing the signal and then applying a classifier. This approach fails to leverage the semantic information in the classifier's output during denoising. In this work, we propose a general, domain-agnostic framework that integrates two interacting diffusion models: one operating on the input signal and the other on the classifier's output logits, without requiring any retraining or fine-tuning of the classifier. This coupled formulation enables mutual guidance, where the enhancing signal refines the class estimation and, conversely, the evolving class logits guide the signal reconstruction towards discriminative regions of the manifold. We introduce three strategies to effectively model the joint distribution of the input and the logit. We evaluated our joint enhancement method for image classification and automatic speech recognition. The proposed framework surpasses traditional sequential enhancement baselines, delivering robust and flexible improvements in classification accuracy under diverse noise conditions.

扩散模型信号增强联合优化分类

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