Data Augmentation Improves OOD Generalization in Dermoscopic Skin Cancer Classification
A new research study on arXiv has highlighted the advantages of data augmentation in improving out-of-domain (OOD) generalization in skin cancer classification using dermoscopy techniques. The team utilized a ConvNeXt-Large architecture and evaluated various augmentation methods on the diverse ISIC Archive dataset, which included Derm7pt. For out-of-domain testing, they incorporated the HAM10000 and ISIC 2019-2020 datasets. Notably, the composite augmentation method outperformed others, particularly with photometric adjustments. The study revealed a significant enhancement in OOD performance, with an average gain of 0.053, suggesting the potential for better classification accuracy of skin cancer types.
Key facts
- Study uses ConvNeXt-Large backbone
- Binary malignant-versus-non-malignant classification
- Data augmentation searched on ISIC Archive and Derm7pt
- OOD test sets: HAM10000 and ISIC 2019-2020
- Mix policy gave largest OOD gain
- Photometric transformations most useful
- OOD gain: +0.053 (95% CI +0.045 to +0.061, p<0.001)
- Consistent across four training seeds
Entities
Institutions
- arXiv