Lightweight classification model training method and device, medium and equipment

The invention provides a lightweight classification model training method and device, a medium and equipment. The method comprises the following steps: collecting first text data with a classification label and second text data without a classification label; according to the first text data, traini...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: ZHANG ZHENG, CHANG BINGXIN, HUANG DENGRONG, YUE AIZHEN
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:The invention provides a lightweight classification model training method and device, a medium and equipment. The method comprises the following steps: collecting first text data with a classification label and second text data without a classification label; according to the first text data, training a large-scale language model to obtain a training convergent large-scale language model; inputting the second text data into the large-scale language model to obtain a classification label corresponding to the second text data, and labeling the second text data by using the classification label; and training a lightweight classification model on the edge device according to the labeled second text data. According to the embodiment of the invention, the problem that more accurate text classification is difficult to realize due to less marked data in a small scene is solved based on the existing large-scale language model. And meanwhile, the lightweight classification model has the advantage of real-time response.