Tree-loss function for training neural networks on weakly-labelled datasets

Sergey Demyanov, Rajib Chakravorty, Zongyuan Ge, Seyedbehzad Bozorgtabar, Michelle Pablo, Adrian Bowling, Rahil Garnavi

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

3 Citations (Scopus)

Abstract

Neural networks are powerful tools for medical image classification and segmentation. However, existing network structures and training procedures assume that the output classes are mutually exclusive and equally important. Many datasets of medical images do not satisfy these conditions. For example, some skin disease datasets have images labelled as coarse-grained class (such as Benign) in addition to images with fine-grained labels (such as a Benign subclass called Blue Nevus), and conventional neural network can not leverage such additional data for training. Also, in the clinical decision making, some classes (such as skin cancer or Melanoma) often carry more importance than other lesion types. We propose a novel Tree-Loss function for training and fine-tuning a neural network classifier using all available labelled images. The key step is the definition of the class taxonomy tree, which is used to describe the relations between labels. The tree can be also adjusted to reflect the desired importance of each class. These steps can be performed by a domain expert without detailed knowledge of machine learning techniques. The experiments demonstrate the improved performance compared with the conventional approach even without using additional data.

Original languageEnglish
Title of host publication2017 IEEE 14th International Symposium on Biomedical Imaging (ISBI 2017)
EditorsSimon Warfield, Arrete Munoz-Barrutia
Place of PublicationPiscataway NJ USA
PublisherIEEE, Institute of Electrical and Electronics Engineers
Pages287-291
Number of pages5
ISBN (Electronic)9781509011728
ISBN (Print)9781509011735
DOIs
Publication statusPublished - 15 Jun 2017
Externally publishedYes
EventIEEE International Symposium on Biomedical Imaging (ISBI) 2017 - Melbourne Convention and Exhibition Centre, Melbourne, Australia
Duration: 18 Apr 201721 Apr 2017
Conference number: 14th
http://biomedicalimaging.org/2017/
https://ieeexplore-ieee-org.ezproxy.lib.monash.edu.au/xpl/conhome/7944115/proceeding

Conference

ConferenceIEEE International Symposium on Biomedical Imaging (ISBI) 2017
Abbreviated titleISBI 2017
CountryAustralia
CityMelbourne
Period18/04/1721/04/17
OtherISBI is a joint initiative from the IEEE Signal Processing Society (SPS) and the IEEE Engineering in Medicine and Biology Society (EMBS). The 2017 meeting will include tutorials, and a scientific program composed of plenary talks, invited special sessions, challenges, as well as oral and poster presentations of peer-reviewed papers.
Internet address

Keywords

  • Classification
  • Loss function
  • Neural networks
  • Skin disease recognition
  • Taxonomy
  • Tree

Cite this