Skip to content

Official implementation of AAAI-21 paper "Label Confusion Learning to Enhance Text Classification Models"

License

Notifications You must be signed in to change notification settings

beyondguo/label_confusion_learning

Repository files navigation

Description:

This is the official implementation of our AAAI-21 accepted paper Label Confusion Learning to Enhance Text Classification Models.

The structure of LCM looks like this:

Here we provide some demo experimental code & datasets.

Environment:

python 3.6 tensorflow 2.2.0 keras 2.3.1

Run a Demo:

LCM-based LSTM:

Run python lcm_exp_on_lstm.py to compare the performance of LSTM, LSTM with label smoothing(LS) and LSTM with LCM.

LCM-based BERT:

Run python lcm_exp_on_bert.py to compare the performance of BERT, BERT with label smoothing(LS) and BERT with LCM.

The final results will be outputted to output/ directory.

The curve below shows our results on 20NG with LSTM as basic predictor. By changing the α, we can control the influence of LCM on the original model.

About

Official implementation of AAAI-21 paper "Label Confusion Learning to Enhance Text Classification Models"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages