CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrieval
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2026
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866911555215425536 |
|---|---|
| author | Li, Xinzhao Vardar, Alptekin Müller, Franz Goli, Navya Tida, Umamaheswara Rao Ni, Kai Hu, Xiaobo Sharon Kämpfe, Thomas Qin, Ruiyang |
| author_facet | Li, Xinzhao Vardar, Alptekin Müller, Franz Goli, Navya Tida, Umamaheswara Rao Ni, Kai Hu, Xiaobo Sharon Kämpfe, Thomas Qin, Ruiyang |
| contents | Deploying Retrieval-Augmented Generation (RAG) on edge devices is in high demand, but is hindered by the latency of massive data movement and computation on traditional architectures. Compute-in-Memory (CiM) architectures address this bottleneck by performing vector search directly within their crossbar structure. However, CiM's adoption for RAG is limited by a fundamental ``representation gap,'' as high-precision, high-dimension embeddings are incompatible with CiM's low-precision, low-dimension array constraints. This gap is compounded by the diversity of CiM implementations (e.g., SRAM, ReRAM, FeFET), each with unique designs (e.g., 2-bit cells, 512x512 arrays). Consequently, RAG data must be naively reshaped to fit each target implementation. Current data shaping methods handle dimension and precision disjointly, which degrades data fidelity. This not only negates the advantages of CiM for RAG but also confuses hardware designers, making it unclear if a failure is due to the circuit design or the degraded input data. As a result, CiM adoption remains limited. In this paper, we introduce CQ-CiM, a unified, hardware-aware data shaping framework that jointly learns Compression and Quantization to produce CiM-compatible low-bit embeddings for diverse CiM designs. To the best of our knowledge, this is the first work to shape data for comprehensive CiM usage on RAG. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2602_20083 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrieval Li, Xinzhao Vardar, Alptekin Müller, Franz Goli, Navya Tida, Umamaheswara Rao Ni, Kai Hu, Xiaobo Sharon Kämpfe, Thomas Qin, Ruiyang Emerging Technologies Hardware Architecture Deploying Retrieval-Augmented Generation (RAG) on edge devices is in high demand, but is hindered by the latency of massive data movement and computation on traditional architectures. Compute-in-Memory (CiM) architectures address this bottleneck by performing vector search directly within their crossbar structure. However, CiM's adoption for RAG is limited by a fundamental ``representation gap,'' as high-precision, high-dimension embeddings are incompatible with CiM's low-precision, low-dimension array constraints. This gap is compounded by the diversity of CiM implementations (e.g., SRAM, ReRAM, FeFET), each with unique designs (e.g., 2-bit cells, 512x512 arrays). Consequently, RAG data must be naively reshaped to fit each target implementation. Current data shaping methods handle dimension and precision disjointly, which degrades data fidelity. This not only negates the advantages of CiM for RAG but also confuses hardware designers, making it unclear if a failure is due to the circuit design or the degraded input data. As a result, CiM adoption remains limited. In this paper, we introduce CQ-CiM, a unified, hardware-aware data shaping framework that jointly learns Compression and Quantization to produce CiM-compatible low-bit embeddings for diverse CiM designs. To the best of our knowledge, this is the first work to shape data for comprehensive CiM usage on RAG. |
| title | CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrieval |
| topic | Emerging Technologies Hardware Architecture |
| url | https://arxiv.org/abs/2602.20083 |