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 language | English (US) |
|---|---|
| Pages (from-to) | 65-80 |
| Number of pages | 16 |
| Journal | Neural Networks |
| Volume | 118 |
| DOIs | |
| State | Published - 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).
| Funders | Funder 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 Foundation | SCH-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 Health | P30EY10572 |
| National Eye Institute and Casey Eye Institute | R01EY019474 |
| 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
- APA
- Standard
- Harvard
- Vancouver
- Author
- BIBTEX
- RIS