arXiv:2502.10648cs.LGstat.ML2025-02被引 21

用大模型指导特征选择,让医学数据建模更可靠。

LLM-Lasso: A Robust Framework for Domain-Informed Feature Selection and Regularization

  • 用大模型从文本中提取领域知识,生成特征惩罚权重
  • 在多个生物医学案例中优于传统Lasso和基线方法
  • 内置验证机制,降低大模型幻觉带来的风险

我们提出LLM-Lasso,一种利用大语言模型(LLMs)引导Lasso ℓ₁回归中特征选择的新框架。不同于仅依赖数值数据的传统方法,LLM-Lasso通过检索增强生成(RAG)管道提取领域知识,实现数据驱动建模与上下文洞察的融合。具体而言,LLM为每个特征生成惩罚因子,并通过可调模型转换为Lasso惩罚权重:领域相关性高的特征获得较低惩罚,更易被保留;相关性低的特征则被施加更高惩罚。该框架具备内部验证机制,动态评估对上下文知识的信任程度,提升鲁棒性,有效缓解大模型可能产生的错误或幻觉。在多个生物医学案例中,LLM-Lasso显著优于标准Lasso及现有特征选择基线,且大模型无需事先访问数据集。据我们所知,这是首个将传统特征选择与基于大模型的领域推理直接结合的方法。

原文摘要 · Abstract (English)

We introduce LLM-Lasso, a novel framework that leverages large language models (LLMs) to guide feature selection in Lasso $\ell_1$ regression. Unlike traditional methods that rely solely on numerical data, LLM-Lasso incorporates domain-specific knowledge extracted from natural language, enhanced through a retrieval-augmented generation (RAG) pipeline, to seamlessly integrate data-driven modeling with contextual insights. Specifically, the LLM generates penalty factors for each feature, which are converted into weights for the Lasso penalty using a simple, tunable model. Features identified as more relevant by the LLM receive lower penalties, increasing their likelihood of being retained in the final model, while less relevant features are assigned higher penalties, reducing their influence. Importantly, LLM-Lasso has an internal validation step that determines how much to trust the contextual knowledge in our prediction pipeline. Hence it addresses key challenges in robustness, making it suitable for mitigating potential inaccuracies or hallucinations from the LLM. In various biomedical case studies, LLM-Lasso outperforms standard Lasso and existing feature selection baselines, all while ensuring the LLM operates without prior access to the datasets. To our knowledge, this is the first approach to effectively integrate conventional feature selection techniques directly with LLM-based domain-specific reasoning.

特征选择大模型应用医学建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。