Domain Knowledge-Enhanced Contrastive Learning for Industry Classification of Enterprises

There exist various online platforms that provide enterprise data and services for e-commerce and foreign trade companies, in which accurate industry classification of enter-prises data is the basis for effective data services. However, there is a pressing need to obtain accurate industry classification automatically. This paper develops a new contrastive learning framework for industry classification of enterprises, which integrates industry domain knowledge with a RoBERTa pretraining model. The effectiveness of our proposed model is demonstrated by extensive experiments on a real enterprise data set. The experimental results show that the model is superior to the baseline and the RoBERTa model with simple contrastive learning method in terms of accuracy, F1 score, recall rate and precision.