In-domain versus out-of-domain transfer learning in plankton image classification.

Andrea Maracani, Vito Paolo Pastore, Lorenzo Natale, Lorenzo Rosasco, Francesca Odone
Author Information
  1. Andrea Maracani: Istituto Italiano di Tecnologia, Genoa, Italy.
  2. Vito Paolo Pastore: MaLGa-DIBRIS, Università degli studi di Genova, Genoa, Italy. Vito.Paolo.Pastore@unige.it.
  3. Lorenzo Natale: Istituto Italiano di Tecnologia, Genoa, Italy.
  4. Lorenzo Rosasco: Istituto Italiano di Tecnologia, Genoa, Italy.
  5. Francesca Odone: MaLGa-DIBRIS, Università degli studi di Genova, Genoa, Italy.

Abstract

Plankton microorganisms play a huge role in the aquatic food web. Recently, it has been proposed to use plankton as a biosensor, since they can react to even minimal perturbations of the aquatic environment with specific physiological changes, which may lead to alterations in morphology and behavior. Nowadays, the development of high-resolution in-situ automatic acquisition systems allows the research community to obtain a large amount of plankton image data. Fundamental examples are the ZooScan and Woods Hole Oceanographic Institution (WHOI) datasets, comprising up to millions of plankton images. However, obtaining unbiased annotations is expensive both in terms of time and resources, and in-situ acquired datasets generally suffer from severe imbalance, with only a few images available for several species. Transfer learning is a popular solution to these challenges, with ImageNet1K being the most-used source dataset for pre-training. On the other hand, datasets like the ZooScan and the WHOI may represent a valuable opportunity to compare out-of-domain and large-scale plankton in-domain source datasets, in terms of performance for the task at hand.In this paper, we design three transfer learning pipelines for plankton image classification, with the aim of comparing in-domain and out-of-domain transfer learning on three popular benchmark plankton datasets. The general framework consists in fine-tuning a pre-trained model on a plankton target dataset. In the first pipeline, the model is pre-trained from scratch on a large-scale plankton dataset, in the second, it is pre-trained on large-scale natural image datasets (ImageNet1K or ImageNet22K), while in the third, a two-stage fine-tuning is implemented (ImageNet [Formula: see text] large-scale plankton dataset [Formula: see text] target plankton dataset). Our results show that an out-of-domain ImageNet22K pre-training outperforms the plankton in-domain ones, with an average boost in test accuracy of around 6%. In the next part of this work, we adopt three ImageNet22k pre-trained Vision Transformers and one ConvNeXt, obtaining results on par (or slightly superior) with the state-of-the-art, corresponding to the usage of CNN models ensembles, with a single model. Finally, we design and test an ensemble of our Vision Transformers and the ConvNeXt, outperforming the state-of-the-art existing works on plankton image classification on the three target datasets. To support scientific community contribution and further research, our implemented code is open-source and available at https://github.com/Malga-Vision/plankton_transfer .

References

  1. Sensors (Basel). 2020 May 28;20(11): [PMID: 32481730]
  2. Front Microbiol. 2021 Nov 15;12:746297 [PMID: 34867861]
  3. BMC Bioinformatics. 2017 Dec 28;18(Suppl 16):570 [PMID: 29297354]
  4. Comput Methods Programs Biomed. 2021 Mar;200:105923 [PMID: 33486341]
  5. Nature. 2010 Jul 29;466(7306):591-6 [PMID: 20671703]
  6. Nature. 2015 May 28;521(7553):436-44 [PMID: 26017442]
  7. Sci Rep. 2020 Jul 22;10(1):12142 [PMID: 32699302]
  8. Science. 2001 Mar 30;291(5513):2594-7 [PMID: 11283369]

Grants

  1. SLING 819789/European Research Council

MeSH Term

Plankton
Deep Learning

Word Cloud

Created with Highcharts 10.0.0planktondatasetsimagedatasetlearningout-of-domainlarge-scalethreepre-trainedin-domaintransferclassificationmodeltargetaquaticmayin-situresearchcommunityZooScanWHOIimagesobtainingtermsavailablepopularImageNet1Ksourcepre-traininghanddesignfine-tuningImageNet22Kimplemented[Formula:seetext]resultstestVisionTransformersConvNeXtstate-of-the-artPlanktonmicroorganismsplayhugerolefoodwebRecentlyproposedusebiosensorsincecanreactevenminimalperturbationsenvironmentspecificphysiologicalchangesleadalterationsmorphologybehaviorNowadaysdevelopmenthigh-resolutionautomaticacquisitionsystemsallowsobtainlargeamountdataFundamentalexamplesWoodsHoleOceanographicInstitutioncomprisingmillionsHoweverunbiasedannotationsexpensivetimeresourcesacquiredgenerallysuffersevereimbalanceseveralspeciesTransfersolutionchallengesmost-usedlikerepresentvaluableopportunitycompareperformancetaskInpaperpipelinesaimcomparingbenchmarkgeneralframeworkconsistsfirstpipelinescratchsecondnaturalthirdtwo-stageImageNetshowoutperformsonesaverageboostaccuracyaround6%nextpartworkadoptImageNet22koneparslightlysuperiorcorrespondingusageCNNmodelsensemblessingleFinallyensembleoutperformingexistingworkssupportscientificcontributioncodeopen-sourcehttps://githubcom/Malga-Vision/plankton_transferIn-domainversus

Similar Articles

Cited By

No available data.