Bg-Mix: On Using Image Backgrounds for Data Augmentation in Electron Microscopy Images
Organizations: Centre for Artificial Intelligence in Public Health Research, Robert Koch Institute, Berlin, Germany · Medical University Lausitz – Carl Thiem, Cottbus, Germany
Abstract
Electron microscopy (EM) serves as a cornerstone in virology enabling the visualization of viral particles and is widely used in research, diagnostics and morphological characterization of pathogens. Computer-assisted image analysis can help facilitate and speed up this labor-intensive method, however, typical EM virus datasets are small and sparsely labeled, resulting in a lack of readily available deep learning models due to training limitations. To overcome data shortages and leverage content-rich, diverse sample backgrounds, we employed background-based data augmentation. To this end, we propose Bg-Mix, a form of data augmentation method, which decomposes training images into foreground objects and backgrounds and subsequently rearranges both into novel augmented images. This not only enhances the performance of object detection models but also increases the diversity of the training data. Using Bg-Mix provides an increase between +0.8 and +3.3 box mean average precision (mAP) points over the baseline when tested across two datasets using two architectures, thereby demonstrating its effectiveness.
Figures & tables
| Model | EMViR | VISN |
|---|---|---|
| box mAP | box mAP | |
| Mask R-CNN | 55.7 | 77.7 |
| + Copy-Paste ( / ) | 55.5 (−0.2) | 75.9 (−1.8) |
| + Bg-Mix ( / ) | 58.3 (+2.6) | 78.5 (+0.8) |
| RTMDet-Ins | 54.5 | 57.9 |
| + Copy-Paste ( / ) | 56.7 (+2.2) | 60.5 (+2.6) |
| Extraction | Blending | Placement | ||||||||||
| Run | A | B | C | D | E | F | G | H | I | J | K | box mAP |
| Baseline | – | – | – | 55.7 | ||||||||
| 1 | ✓ | ✓ | ✓ | 57.7 (+2.0) | ||||||||
| 2 | ✓ | ✓ | ✓ | 57.0 (+1.3) | ||||||||
| 3 | ✓ | ✓ | ✓ | 57.9 (+2.2) | ||||||||
| 4 | ✓ | ✓ | ✓ | 57.5 (+1.8) | ||||||||
Data Augmentation
Data augmentation expands or transforms training examples to improve learning, data efficiency, and robustness to distribution shifts.
See all 266 papers