.png){kind=icon}
Return
Discrimination Neural Network Model for Binary Classification Tasks on Tabular Data
.png){kind=icon}
OA
AI
DOI：10.1109/ACCESS.2023.3243919.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
For the classification task, neural network-based approaches attempt to distinguish between two distributions by determining the joint distribution of input variables for each class. However, the most challenging task is still to classify the observations in the overlapping region of two classes. In this work, we propose a new discrimination neural network (DiscNN) architecture to address this issue. Our DiscNN learns to embed the initial input into more informative representations with better discriminability between the two distributions based on the cosine embedding loss. We also train our proposed model using the few-shot learning method to extract better-generalized representations from the initial input. We applied the DiscNN model to 35 tabular datasets from the OpenML-CC18 benchmark for a binary classification task. Our model showed superior performances on 28 datasets of them. In addition, we also performed experiments on 95 imbalanced datasets from the KEEL repository. The experiment results showed that the DiscNN outperformed the state-of-the-art models, including CatBoost, LightGBM, TabNet, VIME and Scarf, by around 0.23% AUC, 0.20% G-mean, and 1.06% F1 score.
Keywords:
Data models
Task analysis
Predictive models
Neural networks
Training
Loss measurement
Prediction algorithms
Neural network
cosine similarity
classification task
imbalanced problem
tabular data
Journal
IF:
3.6
Papers:
9.8W
Citations:
29.4W
Organization
View more organizations
Cited Papers
ShareSaveNeural Network-Augmented Locally Adaptive Linear Regression Model for Tabular Data.png)
SUSTAINABILITYOAAI
SUSTAINABILITYIF3.3
OAAIShareSave
THE MEANING AND USE OF THE AREA UNDER A RECEIVER OPERATING CHARACTERISTIC (ROC) CURVERADIOLOGY
RADIOLOGYIF15.2
PREAI
ShareSaveGEV-NN: A deep neural network architecture for class imbalance problem in binary classification
PREAI
ShareSaveShareSaveShareSave View more