西安交通大学学报(医学版)2026,Vol.47Issue(3):414-423,10.DOI:10.7652/jdyxb202603003
探索心脏疾病智能预测:人工智能体与文本嵌入模型的综合应用
Exploring heart disease smart prediction:integration of AI agents and text embedding models
摘要
Abstract
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.关键词
大型语言模型/SBERT/OpenAI/人工智能体(AI智能体)/基于智能体的心脏疾病预测模型(ABHDP)Key words
large language model/SBERT/OpenAI/AI agent/agent-based heart disease prediction(ABHDP)model分类
医药卫生引用本文复制引用
徐帆,杨正,周睿,郑莹彬,刘奕杉,舒婷,赵敏..探索心脏疾病智能预测:人工智能体与文本嵌入模型的综合应用[J].西安交通大学学报(医学版),2026,47(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) (No.YLXX24AIA020)