Bayesian Random Semantic Data Augmentation for Medical Image Classification
CoRR(2024)
摘要
Data augmentation is a critical regularization technique for deep neural
networks, particularly in medical image classification. Popular data
augmentation approaches include image transformation-based methods, generative
data augmentation, and automatic data augmentation. However, these approaches
encounter notable limitations: image transformation-based and automated data
augmentation techniques cannot implement semantic transformations, leading to a
constrained variety of augmented samples, and generative data augmentation
methods are computationally expensive. In response to these challenges, we
proposed Bayesian Random Semantic Data Augmentation (BRSDA), a novel,
efficient, and plug-and-play semantic data augmentation method. BRSDA is
motivated by a simple translation in the feature space along specific
directions that can effectuate semantic transformations. When given a feature,
we define its augmentable semantic magnitude as a random variable and estimate
its distribution using variational Bayesian, then sample semantic magnitude and
add to the randomly selected semantic direction to achieve semantic data
augmentation. We demonstrate the effectiveness of BRSDA on five 2D and six 3D
medical image datasets covering nine modalities. We also test BRSDA with
mainstream neural network architectures, showcasing its robustness.
Furthermore, combining BRSDA with other leading data augmentation methods
achieves superior performance. Code is available online at
.
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要