arXiv:2601.20041cs.LGcs.AI2026-01中稿 · ICASSP 2026

让边缘设备上的大模型更准更快地个性化响应。

CiMRAG: CiM-Aware Domain-Adaptive and Noise-Resilient Retrieval-Augmented Generation for Edge-Based LLMs

  • 设计噪声感知嵌入学习框架,适配存内计算硬件
  • 在噪声环境下仍保持高精度检索,提升多场景适应性
  • 适合部署在资源受限的智能终端,如手机、可穿戴设备

基于大语言模型的个性化虚拟助手在边缘设备上日益受到关注,检索增强生成(RAG)通过调用用户个人数据实现定制化响应,成为关键方法。然而,随着用户-模型交互和近期更新数据快速增长,将RAG部署于边缘设备面临效率瓶颈。存内计算(CiM)架构通过原位运算消除内存与计算单元间的数据搬运,缓解此问题,但易受环境噪声影响,降低检索精度。在动态、多领域边缘场景(如旅行、医疗、法律)中,准确性和适应性至关重要。为此,我们提出任务导向的噪声鲁棒嵌入学习(TONEL),提升RAG在噪声边缘环境中的鲁棒性与领域自适应能力。TONEL采用噪声感知投影模型,学习符合CiM硬件约束的任务特定嵌入,实现在噪声条件下的精准检索。在个性化基准上的大量实验表明,该方法相对于强基线具有显著优势,尤其在任务特定噪声场景中表现突出。

原文摘要 · Abstract (English)

Personalized virtual assistants powered by large language models (LLMs) on edge devices are attracting growing attention, with Retrieval-Augmented Generation (RAG) emerging as a key method for personalization by retrieving relevant profile data and generating tailored responses. However, deploying RAG on edge devices faces efficiency hurdles due to the rapid growth of profile data, such as user-LLM interactions and recent updates. While Computing-in-Memory (CiM) architectures mitigate this bottleneck by eliminating data movement between memory and processing units via in-situ operations, they are susceptible to environmental noise that can degrade retrieval precision. This poses a critical issue in dynamic, multi-domain edge-based scenarios (e.g., travel, medicine, and law) where both accuracy and adaptability are paramount. To address these challenges, we propose Task-Oriented Noise-resilient Embedding Learning (TONEL), a framework that improves noise robustness and domain adaptability for RAG in noisy edge environments. TONEL employs a noise-aware projection model to learn task-specific embeddings compatible with CiM hardware constraints, enabling accurate retrieval under noisy conditions. Extensive experiments conducted on personalization benchmarks demonstrate the effectiveness and practicality of our methods relative to strong baselines, especially in task-specific noisy scenarios.

边缘计算RAG存内计算噪声鲁棒

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