Radar Live
✦ Ask AI
AI INTELLIGENCE & SIGNALS

ARM: Attention with Routed-Memory for Learnable Sparse Control

Phân tích đã xử lý trước và các nguồn liên quan.

Tín hiệu Radar

65 RADAR
Research KV Cache Optimization · 1 nguồn đối chiếu · 2026-09-21 11:11:17 +0000 UTC

ARM: Attention with Routed-Memory for Learnable Sparse Control

Nguồn: arXiv AI
↗

ARM (Attention with Routed-Memory) là kiến trúc bộ nhớ KV cache kích thước cố định, có thể vi phân hoàn toàn, được tổ chức dưới dạng bộ định tuyến phân cấp. Bằng cách sử dụng Gumbel-Softmax và cổng cập nhật sigmoid, ARM kết hợp mềm thông tin mới vào bộ nhớ thay vì loại bỏ token cứng, giúp hạn chế tối đa mất mát dữ liệu. Phương pháp này còn tích hợp một policy cho phép truy cập linh hoạt lượng bộ nhớ tùy theo độ phức tạp của đầu vào, tối ưu hiệu quả suy luận cho cả ngữ cảnh ngắn và dài.

✦ TÁC ĐỘNG & GIÁ TRỊ THỰC TIỄN

Dung lượng KV cache tăng tuyến tính theo chiều dài chuỗi là rào cản kỹ thuật lớn nhất trong việc mở rộng suy luận ngữ cảnh dài của LLM. ARM mang đến giải pháp đột phá khi biến KV cache thành bộ nhớ có thể học được với kích thước cố định và cập nhật mềm vi phân, giúp loại bỏ hiện tượng tràn bộ nhớ mà vẫn bảo toàn thông tin quan trọng và tối ưu độ trễ suy luận.

Developer 69
Business 71
Novelty 82
Actionable 65

Nội dung thu thập đã chuẩn hóa

Collected Evidence

Nội dung văn bản được dùng làm dữ liệu đối chứng cho mô hình AI, không phải chỉ thị hệ thống.

Despite advances in long-context inference, large language models (LLMs) remain fundamentally limited by the key-value (KV) caching mechanisms that are necessary for stable computation. Techniques such as selective token eviction and pruning have vastly mitigated these issues, but often discard core information to manage the growing cache. In this paper, we propose Attention with Routed Memory (ARM) a novel KV caching structure that introduces a fully differentiable, fixed-size memory system organized as a hierarchical router. Via a Gumbel-Softmax, ARM learns to select memory slots and perform sigmoid-gated updates that softly combine new and stored information, avoiding hard eviction and reducing information loss. By further training a policy to dynamically select varying amounts of memory at inference, ARM adapts its accesses for both simple contexts and inputs that require deeper reasoning, enabling more scalable and effective retrieval on both short- and long-contexts. Experimental results on standard commonsense and long-context reasoning benchmarks demonstrate that ARM achieves superior performance and efficiency compared to fixed KV-caching approaches, while remaining efficient and scalable in terms of both memory and generation latency.

Các nguồn đối chiếu cho sự kiện này

1 nguồn
Thông tin phân tích AI & Model Details ▾
Provider: openai-compatible · Model: gemini-3.8-flash-high · Version: analysis-v1 · Time: 2026-09-22 09:49:37 +0000 UTC
{"tags": ["KV Cache", "Attention Mechanism", "Efficient Inference", "Long-Context LLMs", "Sparse Attention", "Gumbel-Softmax"], "risks": ["Đòi hỏi phải huấn luyện mô hình kèm policy định tuyến, không thể áp dụng trực tiếp dưới dạng plug-and-play cho các mô hình đã huấn luyện sẵn mà không đào tạo lại.", "Yêu cầu tối ưu hóa kernel tùy biến (custom kernels/CUDA) để đạt hiệu năng phần cứng tối đa trong môi trường production thực tế.", "Cơ chế nén và cập nhật mềm (soft combination) có thể làm suy hao các chi tiết chính xác tuyệt đối trong các tác vụ truy xuất thông tin cụ thể (needle-in-a-haystack)."], "category": "Research", "entities": ["ARM", "Attention with Routed-Memory", "Gumbel-Softmax", "LLMs"], "summary_vi": "ARM (Attention with Routed-Memory) là kiến trúc bộ nhớ KV cache kích thước cố định, có thể vi phân hoàn toàn, được tổ chức dưới dạng bộ định tuyến phân cấp. Bằng cách sử dụng Gumbel-Softmax và cổng cập nhật sigmoid, ARM kết hợp mềm thông tin mới vào bộ nhớ thay vì loại bỏ token cứng, giúp hạn chế tối đa mất mát dữ liệu. Phương pháp này còn tích hợp một policy cho phép truy cập linh hoạt lượng bộ nhớ tùy theo độ phức tạp của đầu vào, tối ưu hiệu quả suy luận cho cả ngữ cảnh ngắn và dài.", "key_changes": ["Đề xuất kiến trúc Attention with Routed-Memory (ARM) sử dụng hệ thống bộ nhớ kích thước cố định, có thể vi phân hoàn toàn và tổ chức dưới dạng bộ định tuyến phân cấp (hierarchical router).", "Ứng dụng Gumbel-Softmax để lựa chọn slot bộ nhớ và thực hiện cập nhật có cổng sigmoid (sigmoid-gated updates), kết hợp mềm thông tin mới và cũ thay vì loại bỏ cứng (hard eviction).", "Tích hợp cơ chế huấn luyện policy để tự động điều chỉnh linh hoạt lượng bộ nhớ truy cập tại thời điểm suy luận tùy theo độ phức tạp của ngữ cảnh.", "Đạt hiệu năng vượt trội trên các benchmark suy luận ngữ cảnh dài và commonsense, đồng thời giảm dung lượng bộ nhớ lẫn độ trễ sinh từ (latency)."], "sub_category": "KV Cache Optimization", "novelty_score": 82, "business_score": 75, "research_score": 88, "why_it_matters": "Dung lượng KV cache tăng tuyến tính theo chiều dài chuỗi là rào cản kỹ thuật lớn nhất trong việc mở rộng suy luận ngữ cảnh dài của LLM. ARM mang đến giải pháp đột phá khi biến KV cache thành bộ nhớ có thể học được với kích thước cố định và cập nhật mềm vi phân, giúp loại bỏ hiện tượng tràn bộ nhớ mà vẫn bảo toàn thông tin quan trọng và tối ưu độ trễ suy luận.", "developer_score": 72, "importance_score": 80, "possible_use_cases": ["Tối ưu hóa dung lượng VRAM cho hạ tầng serving các mô hình ngôn ngữ lớn (LLM) phục vụ văn bản dài.", "Cải thiện hiệu năng xử lý ngữ cảnh dài cho các hệ thống AI Agent cần duy trì lịch sử hội thoại hoặc tài liệu lớn liên tục.", "Giảm độ trễ sinh từ và chi phí vận hành phần cứng trong các ứng dụng tra cứu, phân tích tài liệu phức tạp."], "actionability_score": 65}
← Trang trước Trang 1