Multiple Instance Pathology Image Diagnosis Model based on Channel Attention and Data Augmentation

Abstract

The application of machine learning in the medical field has resulted in significant advancements in computer-aided pathological diagnosis. Multiple instance learning (MIL) has emerged as a promising approach for pathological image classification, particularly in scenarios where local annotations are lacking. However, current MIL models often overlook the importance of feature weights in the channel dimension and struggle with imbalanced positive and negative data. To address these limitations, an integration of a channel attention (CA) module and an augmented data (AUG) mechanism into the MIL model is proposed, resulting in improved performance. The CA module dynamically assigns weights to example features in the channel dimension, enhancing or suppressing features adaptively. Additionally, the AUG mechanism effectively balances the distribution of positive and negative data, significantly reducing false negatives. Through ablation experiments, the contributions of the CA module and AUG mechanism in enhancing the overall model performance are analyzed. Experimental validations on the CAMELYON16/17 public pathological image datasets demonstrate that the proposed model and method outperform existing approaches, with particular emphasis on reducing false negatives.