arXiv:2505.13039cs.CV2025-05被引 4

用多尺度感受野模拟专家会诊,提升医学图像分类的准确与公平性。

Expert-Like Reparameterization of Heterogeneous Pyramid Receptive Fields in Efficient CNNs for Fair Medical Image Classification

  • 设计异构金字塔感受野,通过多种核大小捕捉病灶多样特征。
  • 在多个数据集上实现更高分类精度与更优公平性,计算开销可控。
  • 适合追求模型公平性与高精度的医疗影像分析研究者使用。

高效卷积神经网络(CNN)架构设计受到广泛关注。然而,现有方法通常采用单一感受野、小尺寸非对称感受野或金字塔感受野来学习特征表示,在医学图像分类任务中仍面临两大挑战:1)难以高效捕捉多样病变特征,如微小、位置特定、小而显著的病灶,这些特征在不平衡医学图像分类中起关键作用;2)模型预测常存在偏差,实际诊疗中风险高。为此,本文提出专家式异构金字塔感受野重参数化(ERoHPRF),通过设计异构金字塔感受野组合,模仿多专家会诊模式,利用多尺度卷积操作有效捕捉不同重要性的病灶特征。同时,引入专家式结构重参数化技术,采用两阶段策略合并参数,在保持与单感受野相当计算成本和推理速度的前提下,实现性能突破。将该方法集成至主流高效CNN架构后,大量实验表明,其在医学图像分类、公平性与计算开销间取得更优平衡,显著优于当前先进方法。代码已开源:https://github.com/XiaoLing12138/Expert-Like-Reparameterization-of-Heterogeneous-Pyramid-Receptive-Fields。

原文摘要 · Abstract (English)

Efficient convolutional neural network (CNN) architecture design has attracted growing research interests. However, they typically apply single receptive field (RF), small asymmetric RFs, or pyramid RFs to learn different feature representations, still encountering two significant challenges in medical image classification tasks: 1) They have limitations in capturing diverse lesion characteristics efficiently, e.g., tiny, coordination, small and salient, which have unique roles on the classification results, especially imbalanced medical image classification. 2) The predictions generated by those CNNs are often unfair/biased, bringing a high risk when employing them to real-world medical diagnosis conditions. To tackle these issues, we develop a new concept, Expert-Like Reparameterization of Heterogeneous Pyramid Receptive Fields (ERoHPRF), to simultaneously boost medical image classification performance and fairness. This concept aims to mimic the multi-expert consultation mode by applying the well-designed heterogeneous pyramid RF bag to capture lesion characteristics with varying significances effectively via convolution operations with multiple heterogeneous kernel sizes. Additionally, ERoHPRF introduces an expert-like structural reparameterization technique to merge its parameters with the two-stage strategy, ensuring competitive computation cost and inference speed through comparisons to a single RF. To manifest the effectiveness and generalization ability of ERoHPRF, we incorporate it into mainstream efficient CNN architectures. The extensive experiments show that our proposed ERoHPRF maintains a better trade-off than state-of-the-art methods in terms of medical image classification, fairness, and computation overhead. The code of this paper is available at https://github.com/XiaoLing12138/Expert-Like-Reparameterization-of-Heterogeneous-Pyramid-Receptive-Fields.

医学影像公平性高效网络感受野

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