探索心脏疾病智能预测:人工智能体与文本嵌入模型的综合应用OA
Exploring heart disease smart prediction:integration of AI agents and text embedding models
目的 探讨人工智能体(AI智能体)如何整合多种分类算法和文本嵌入模型,以优化心脏疾病诊断中的预测准确性和上下文相关性.方法 基于智能体的心脏疾病预测模型(agent-based heart disease prediction,ABHDP)整合了分类建模、文档检索和用户交互组件.研究利用心脏疾病临床数据集以及补充数据集,训练了包括随机森林和梯度提升在内的多种分类算法,文本嵌入模型以及本地化部署的DeepSeek.采用准确率、精确率、召回率、F1值和受试者工作特征(receiver operating characteristic,ROC)曲线分析量化模型性能和效果.结果 对分类和连续变量的分析揭示了与心脏疾病患病率相关的重要模式.CatBoost和随机森林等预测模型显示出较高的性能,召回率分别为0.980和0.979.在文本嵌入模型方面,FastText的F1评分为46%,而SBERT和 OpenAI嵌入模型的表现优于Fast-Text,分别达到了95%和96%的F1评分.结论 ABHDP模型可提高心脏疾病管理的准确性和个性化水平,推动精准医疗的发展.
Objective To develop an agent-based heart disease prediction(ABHDP)model that combines multiple machine learning classifiers with text embedding model to improve prediction performance and contextual interpretation in heart disease risk assessment.Methods The proposed ABHDP model integrates three functional modules:disease classification,medical knowledge retrieval,and interactive interpretation.A heart disease dataset was used together with medical knowledge dataset.Eleven classification algorithms and three text embedding approaches were evaluated.Model performance was evaluated using accuracy,precision,recall,F1-score,and receiver operating characteristic(ROC).Results Analysis of categorical and continuous variables revealed important patterns associated with the prevalence of heart disease.Among the classification algorithms,CatBoost and random forest achieved the highest recall values(0.980 and 0.979,respectively).For text embedding methods,FastText produced an F1-score of 46%,while SBERT and OpenAI achieved substantially higher scores of 95%and 96%.Conclusion The proposed ABHDP model demonstrates the potential to support interpretable and intelligent clinical decision-making for heart disease risk prediction,thus promoting the development of precision healthcare.
徐帆;杨正;周睿;郑莹彬;刘奕杉;舒婷;赵敏
国家卫生健康委医院管理研究所,北京 100044国家健康医疗大数据研究院(深圳)工程技术研究中心,广东 深圳 518172香港中文大学(深圳)附属第二医院 & 深圳市龙岗区人民医院网络中心,广东 深圳 518172厦门大学附属第一医院信息部,福建 厦门 361003太原理工大学软件学院,山西 太原 030024国家卫生健康委统计信息中心调查评价处,北京 100810厦门市杏林医院,福建 厦门 361003
医药卫生
大型语言模型SBERTOpenAI人工智能体(AI智能体)基于智能体的心脏疾病预测模型(ABHDP)
large language modelSBERTOpenAIAI agentagent-based heart disease prediction(ABHDP)model
《西安交通大学学报(医学版)》 2026 (3)
414-423,10
国家卫生健康委医院管理研究所医疗人工智能临床应用研究课题(No.YLXX24AIA020)Supported by the National Health Commission Hospital Management Institute's Research Topic on Clinical Application of Medical Artificial Intelligence(No.YLXX24AIA020)
评论