arXiv:2409.14313cs.CV2024-09

针对长尾分布图像分类,提出非各向同性扩散模型提升罕见类识别准确率。

Anisotropic Diffusion Probabilistic Model for Imbalanced Image Classification

  • 根据误差分析理论设计不同类别噪声水平,控制扩散速度差异
  • 在皮肤病变数据集上,稀有类F1分数提升4%和3%以上
  • 融合全局与局部图像先验,增强空间判别力,适合医疗图像分析

真实世界数据常呈长尾分布,尾部样本稀缺严重影响模型泛化能力。基于随机微分方程的去噪扩散概率模型(DDPM)在图像分类中表现优异,但在尾部类别上表现不佳。本文提出非各向同性扩散概率模型(ADPM)解决不平衡分类问题。通过利用数据分布调控不同类别样本在前向过程中的扩散速度,有效提升反向过程中去噪器对尾部类别的分类精度。具体地,基于误差分析理论提出类别相关的噪声水平选择策略;同时在前向过程融合全局与局部图像先验以增强空间判别能力,在反向过程引入语义级上下文信息以提升模型判别力与鲁棒性。在四个医学基准数据集上的实验验证了方法有效性。结果表明,该模型显著提升稀有类分类准确率,且保持头部类性能。在皮肤病变数据集PAD-UFES和HAM10000上,F1得分分别较原始扩散模型提升4%和3%。

原文摘要 · Abstract (English)

Real-world data often has a long-tailed distribution, where the scarcity of tail samples significantly limits the model's generalization ability. Denoising Diffusion Probabilistic Models (DDPM) are generative models based on stochastic differential equation theory and have demonstrated impressive performance in image classification tasks. However, existing diffusion probabilistic models do not perform satisfactorily in classifying tail classes. In this work, we propose the Anisotropic Diffusion Probabilistic Model (ADPM) for imbalanced image classification problems. We utilize the data distribution to control the diffusion speed of different class samples during the forward process, effectively improving the classification accuracy of the denoiser in the reverse process. Specifically, we provide a theoretical strategy for selecting noise levels for different categories in the diffusion process based on error analysis theory to address the imbalanced classification problem. Furthermore, we integrate global and local image prior in the forward process to enhance the model's discriminative ability in the spatial dimension, while incorporate semantic-level contextual information in the reverse process to boost the model's discriminative power and robustness. Through comparisons with state-of-the-art methods on four medical benchmark datasets, we validate the effectiveness of the proposed method in handling long-tail data. Our results confirm that the anisotropic diffusion model significantly improves the classification accuracy of rare classes while maintaining the accuracy of head classes. On the skin lesion datasets, PAD-UFES and HAM10000, the F1-scores of our method improved by 4% and 3%, respectively compared to the original diffusion probabilistic model.

图像分类长尾分布扩散模型医疗图像

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