English

Scalable Twin Neural Networks for Classification of Unbalanced Data

Machine Learning 2019-02-12 v2

Abstract

Twin Support Vector Machines (TWSVMs) have emerged an efficient alternative to Support Vector Machines (SVM) for learning from imbalanced datasets. The TWSVM learns two non-parallel classifying hyperplanes by solving a couple of smaller sized problems. However, it is unsuitable for large datasets, as it involves matrix operations. In this paper, we discuss a Twin Neural Network (Twin NN) architecture for learning from large unbalanced datasets. The Twin NN also learns an optimal feature map, allowing for better discrimination between classes. We also present an extension of this network architecture for multiclass datasets. Results presented in the paper demonstrate that the Twin NN generalizes well and scales well on large unbalanced datasets.

Keywords

Cite

@article{arxiv.1705.00347,
  title  = {Scalable Twin Neural Networks for Classification of Unbalanced Data},
  author = {Jayadeva and Himanshu Pant and Sumit Soman and Mayank Sharma},
  journal= {arXiv preprint arXiv:1705.00347},
  year   = {2019}
}

Comments

20 pages, 8 figures, 14 tables