用知识蒸馏压缩BERT模型,高效识别社交网络中注意力缺陷症状严重程度
Larger models yield better results? Streamlined severity classification of ADHD-related concerns using BERT-based knowledge distillation
- 通过知识蒸馏构建轻量级学生模型LastBERT,参数从1.1亿减至2900万
- 在真实ADHD数据集上达到85%准确率和F1分数,性能接近临床专用模型
- 模型体积缩小73.64%,适合在资源受限设备如Google Colab上部署
本研究聚焦知识蒸馏在生成轻量但强大的BERT类自然语言处理模型中的效率。创建模型后,将所得模型LastBERT应用于从社交媒体文本中分类注意力缺陷多动障碍(ADHD)相关问题严重程度的真实任务。相比原始的BERT base模型(1.1亿参数),定制化的学生模型将参数降至2900万,模型大小减少约73.64%。在包含释义识别、情感分析和文本分类的GLUE基准测试中,该学生模型在多个任务上仍保持强劲表现。在真实世界ADHD数据集上,模型取得85%的准确率和F1分数。与DistilBERT(6600万参数)和ClinicalBERT(1.1亿参数)相比,其性能相当:DistilBERT略优(87%),ClinicalBERT为86%。结果表明,LastBERT能有效分类ADHD严重程度,为心理健康从业者提供评估社交平台用户内容的实用工具。研究凸显知识蒸馏在资源受限条件下构建高效模型的潜力,推动NLP与精神健康诊断的落地应用。模型尺寸显著缩减且性能损失不明显,大幅降低训练与部署所需的计算资源,提升可及性,尤其适配Google Colab等通用计算平台。
原文摘要 · Abstract (English)
This work focuses on the efficiency of the knowledge distillation approach in generating a lightweight yet powerful BERT based model for natural language processing applications. After the model creation, we applied the resulting model, LastBERT, to a real-world task classifying severity levels of Attention Deficit Hyperactivity Disorder (ADHD)-related concerns from social media text data. Referring to LastBERT, a customized student BERT model, we significantly lowered model parameters from 110 million BERT base to 29 million, resulting in a model approximately 73.64% smaller. On the GLUE benchmark, comprising paraphrase identification, sentiment analysis, and text classification, the student model maintained strong performance across many tasks despite this reduction. The model was also used on a real-world ADHD dataset with an accuracy and F1 score of 85%. When compared to DistilBERT (66M) and ClinicalBERT (110M), LastBERT demonstrated comparable performance, with DistilBERT slightly outperforming it at 87%, and ClinicalBERT achieving 86% across the same metrics. These findings highlight the LastBERT model's capacity to classify degrees of ADHD severity properly, so it offers a useful tool for mental health professionals to assess and comprehend material produced by users on social networking platforms. The study emphasizes the possibilities of knowledge distillation to produce effective models fit for use in resource-limited conditions, hence advancing NLP and mental health diagnosis. Furthermore underlined by the considerable decrease in model size without appreciable performance loss is the lower computational resources needed for training and deployment, hence facilitating greater applicability. Especially using readily available computational tools like Google Colab. This study shows the accessibility and usefulness of advanced NLP methods in pragmatic world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。