提升生成式AI通信中令牌传输的抗丢包能力
Low-Complexity Semantic Packet Aggregation for Token Communication via Lookahead Search
- 用残差语义分数替代整体相似度,实现语义鲁棒性
- 通过前瞻搜索策略使计算复杂度降低40倍
- 适合远程AI内容生成与无线大模型应用
令牌是生成式AI和大语言模型的基本处理单元,令牌通信对实现远程AI生成内容及无线大模型应用至关重要。由于令牌语义依赖上下文,单个令牌丢失可能导致语义严重失真。为此,本文提出一种基于前瞻搜索的语义包聚合框架(SemPA-Look),通过引入令牌级残差语义分数(RSS)作为消息级平均令牌相似度(ATS)的代理指标,增强语义保真度;同时采用固定深度无放回采样、固定宽度有放回采样的前瞻搜索算法,实现线性复杂度。在MS-COCO数据集上的远程AIGC任务实验表明,该方法在保持与穷举搜索相当的高ATS和LPIPS得分的同时,计算复杂度降低达40倍;相比遗传算法等线性复杂度方法,复杂度更低10倍,适用于远程AIGC及其他令牌通信场景。
原文摘要 · Abstract (English)
Tokens are fundamental processing units of generative AI (GenAI) and large language models (LLMs), and token communication (TC) is essential for enabling remote AI-generate content (AIGC) and wireless LLM applications. Unlike traditional bits, each of which is independently treated, the semantics of each token depends on its surrounding context tokens. This inter-token dependency makes TC vulnerable to outage channels, where the loss of a single token can significantly distort the original message semantics. Motivated by this, this paper focuses on optimizing token packetization to maximize the average token similarity (ATS) between the original and received token messages under outage channels. Due to inter-token dependency, this token grouping problem is combinatorial, with complexity growing exponentially with message length. To address this, we propose a novel framework of semantic packet aggregation with lookahead search (SemPA-Look), built on two core ideas. First, it introduces the residual semantic score (RSS) as a token-level surrogate for the message-level ATS, allowing robust semantic preservation even when a certain token packet is lost. Second, instead of full search, SemPA-Look applies a lookahead search-inspired algorithm that samples intra-packet token candidates without replacement (fixed depth), conditioned on inter-packet token candidates sampled with replacement (fixed width), thereby achieving linear complexity. Experiments on a remote AIGC task with the MS-COCO dataset (text captioned images) demonstrate that SemPA-Look achieves high ATS and LPIPS scores comparable to exhaustive search, while reducing computational complexity by up to 40$\times$. Compared to other linear-complexity algorithms such as the genetic algorithm (GA), SemPA-Look achieves 10$\times$ lower complexity, demonstrating its practicality for remote AIGC and other TC applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。