Transformer-Based Classification of User Queries for Medical Consultancy

Automation and Remote Control, США. 2024. Vol. 85. No. 3. P. 297–308.

Lyutkin D. A., D. V. Pozdnyakov, Soloviev A. A., Zhukov D. V., Malik M. S., D. I. Ignatov

The need for skilled medical support is growing in the era of digital healthcare. This research presents an innovative strategy, utilizing the RuBERT model, for categorizing user inquiries in the field of medical consultation with a focus on expert specialization. By harnessing the capabilities of transformers, we fine-tuned the pretrained RuBERT model on a varied dataset, which facilitates precise correspondence between queries and particular medical specialisms. Using a comprehensive dataset, we have demonstrated our approach’s superior performance with an Fl-score of over 91.8%, calculated through both cross-validation and the traditional split of test and train datasets. Our approach has shown excellent generalization across medical domains such as cardiology, neurology and dermatology. This methodology provides practical benefits by directing users to appropriate specialists for prompt and targeted medical advice. It also enhances healthcare system efficiency, reduces practitioner burden, and improves patient care quality. In summary, our suggested strategy facilitates the attainment of specific medical knowledge, offering prompt and precise advice within the digital healthcare field.

Keywords: трансформеры, Transformers, Russian medical texts, Many-class Learning, многоклассовое обучение, Российские медицинские тексты

Publication based on the results of: Models and methods for natural language processing, recommendation systems and data mining (2024)