Back to Search Start Over

Twin Neural Networks for the classification of large unbalanced datasets

Authors :
Mayank Sharma
Sumit Soman
Himanshu Pant
Jayadeva
Source :
Neurocomputing. 343:34-49
Publication Year :
2019
Publisher :
Elsevier BV, 2019.

Abstract

Twin Support Vector Machines (TWSVMs) have emerged as 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 objective functions of the networks in the Twin NN are designed to realize the idea of the Twin SVM with non-parallel decision boudaries for the respective classes, while also being able to reduce model complexity. The Twin NN optimizes the feature map, allowing for better discrimination between classes. The paper also discusses an extension of the Twin NN for multiclass datasets. This architecture trains as many neural networks as the number of classes, and has the additional advantage that it does not have any hyper-parameter which requires tuning. Results presented in the paper demonstrate that the Twin NN generalizes well and scales well on large unbalanced datasets.

Details

ISSN :
09252312
Volume :
343
Database :
OpenAIRE
Journal :
Neurocomputing
Accession number :
edsair.doi...........efa565b0a39aac65c38f63cfd60e4dd0