arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.12843cs.LGeess.SPstat.ML
收藏

EQ-Net:一个用于对数似然比估计和Quantization的统一深度学习框架

EQ-Net: A Unified Deep Learning Framework for Log-Likelihood Ratio Estimation and Quantization

The University of Texas at Austin(德克萨斯大学奥斯汀分校)

Marius Arvinte, Ahmed H. Tewfik, Sriram Vishwanath

摘要:

在这项工作中,我们介绍了EQ-Net:第一个整体框架,它使用数据驱动的方法解决了对数似然比(LLR)估计和量化的任务。我们激励我们的方法与理论见解的两个实际的估计算法的复杂性频谱的两端,并揭示了算法的复杂性和信息瓶颈方法之间的连接:简单的算法承认较小的瓶颈时,代表他们的解决方案。这促使我们提出一个两阶段的算法,使用LLR压缩作为借口任务的估计,并专注于低延迟,高性能的实现,通过深神经网络。我们进行了广泛的实验评估和证明,我们的单一架构实现了国家的最先进的结果在这两个任务相比,以前的方法,与增益的量化效率高达$20\%$和减少估计延迟时,测量通用和图形处理单元(GPU)的$60\%$。特别是,我们的方法减少了GPU的推理延迟的两倍以上,在几个多输入多输出(MIMO)配置。最后,我们证明了我们的方案对分布偏移具有鲁棒性,并且在5G信道模型以及信道估计误差上进行评估时保留了其性能的重要部分。

英文摘要:

In this work, we introduce EQ-Net: the first holistic framework that solves both the tasks of log-likelihood ratio (LLR) estimation and quantization using a data-driven method. We motivate our approach with theoretical insights on two practical estimation algorithms at the ends of the complexity spectrum and reveal a connection between the complexity of an algorithm and the information bottleneck method: simpler algorithms admit smaller bottlenecks when representing their solution. This motivates us to propose a two-stage algorithm that uses LLR compression as a pretext task for estimation and is focused on low-latency, high-performance implementations via deep neural networks. We carry out extensive experimental evaluation and demonstrate that our single architecture achieves state-of-the-art results on both tasks when compared to previous methods, with gains in quantization efficiency as high as $20\%$ and reduced estimation latency by up to $60\%$ when measured on general purpose and graphical processing units (GPU). In particular, our approach reduces the GPU inference latency by more than two times in several multiple-input multiple-output (MIMO) configurations. Finally, we demonstrate that our scheme is robust to distributional shifts and retains a significant part of its performance when evaluated on 5G channel models, as well as channel estimation errors.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑