Skip to main navigation Skip to search Skip to main content

Classification and comparison via neural networks

  • İlkay Yıldız
  • , Peng Tian
  • , Jennifer Dy
  • , Deniz Erdoğmuş
  • , James Brown
  • , Jayashree Kalpathy-Cramer
  • , Susan Ostmo
  • , J. Peter Campbell
  • , Michael F. Chiang
  • , Stratis Ioannidis

Research output: Contribution to journalArticlepeer-review

Abstract

We consider learning from comparison labels generated as follows: given two samples in a dataset, a labeler produces a label indicating their relative order. Such comparison labels scale quadratically with the dataset size; most importantly, in practice, they often exhibit lower variance compared to class labels. We propose a new neural network architecture based on siamese networks to incorporate both class and comparison labels in the same training pipeline, using Bradley–Terry and Thurstone loss functions. Our architecture leads to a significant improvement in predicting both class and comparison labels, increasing classification AUC by as much as 35% and comparison AUC by as much as 6% on several real-life datasets. We further show that, by incorporating comparisons, training from few samples becomes possible: a deep neural network of 5.9 million parameters trained on 80 images attains a 0.92 AUC when incorporating comparisons.

Original languageEnglish (US)
Pages (from-to)65-80
Number of pages16
JournalNeural Networks
Volume118
DOIs
StatePublished - Oct 1 2019

Funding

Our work is supported by NIH, USA ( R01EY019474 , P30EY10572 ), National Science Foundation, USA ( SCH-1622542 at MGH; SCH-1622536 at Northeastern; SCH-1622679 at OHSU), and by unrestricted departmental funding from Research to Prevent Blindness, USA (OHSU).

FundersFunder number
NORTHEASTERN UNIVERSITY
Author National Science Foundation National Science Foundation National Institutes of Health National Institutes of Health National Institutes of Health National Institutes of Health National Science Foundation National Science FoundationSCH-1622542, SCH-1622536, SCH-1622679
Author National Institutes of Health National Institutes of Health National Institutes of Health National Institutes of Health The Bev Hartig Huntington's Disease Foundation National Institutes of HealthP30EY10572
National Eye Institute and Casey Eye InstituteR01EY019474
Research to Prevent Blindness
Oregon State University/Oregon Health and Science University

    Keywords

    • Classification
    • Comparison
    • Joint learning
    • Neural network
    • Siamese network

    ASJC Scopus subject areas

    • Cognitive Neuroscience
    • Artificial Intelligence

    Fingerprint

    Dive into the research topics of 'Classification and comparison via neural networks'. Together they form a unique fingerprint.

    Cite this