The Pitfalls of Sample Selection: A Case Study on Lung Nodule Classification

Vasileios Baltatzis, Kyriaki Margarita Bintsi, Loïc Le Folgoc, Octavio E. Martinez Manzanera, Sam Ellis, Arjun Nair, Sujal Desai, Ben Glocker, Julia A. Schnabel

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Scopus citations

Abstract

Using publicly available data to determine the performance of methodological contributions is important as it facilitates reproducibility and allows scrutiny of the published results. In lung nodule classification, for example, many works report results on the publicly available LIDC dataset. In theory, this should allow a direct comparison of the performance of proposed methods and assess the impact of individual contributions. When analyzing seven recent works, however, we find that each employs a different data selection process, leading to largely varying total number of samples and ratios between benign and malignant cases. As each subset will have different characteristics with varying difficulty for classification, a direct comparison between the proposed methods is thus not always possible, nor fair. We study the particular effect of truthing when aggregating labels from multiple experts. We show that specific choices can have severe impact on the data distribution where it may be possible to achieve superior performance on one sample distribution but not on another. While we show that we can further improve on the state-of-the-art on one sample selection, we also find that on a more challenging sample selection, on the same database, the more advanced models underperform with respect to very simple baseline methods, highlighting that the selected data distribution may play an even more important role than the model architecture. This raises concerns about the validity of claimed methodological contributions. We believe the community should be aware of these pitfalls and make recommendations on how these can be avoided in future work.

Original languageEnglish
Title of host publicationPredictive Intelligence in Medicine - 4th International Workshop, PRIME 2021, Held in Conjunction with MICCAI 2021, Proceedings
EditorsIslem Rekik, Ehsan Adeli, Sang Hyun Park, Julia Schnabel
PublisherSpringer Science and Business Media Deutschland GmbH
Pages201-211
Number of pages11
ISBN (Print)9783030876012
DOIs
StatePublished - 2021
Event4th International Workshop on Predictive Intelligence in Medicine, PRIME 2021, held in conjunction with 24th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2021 - Virtual, Online
Duration: 1 Oct 20211 Oct 2021

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12928 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference4th International Workshop on Predictive Intelligence in Medicine, PRIME 2021, held in conjunction with 24th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2021
CityVirtual, Online
Period1/10/211/10/21

Fingerprint

Dive into the research topics of 'The Pitfalls of Sample Selection: A Case Study on Lung Nodule Classification'. Together they form a unique fingerprint.

Cite this