FLASH Viterbi: Fast and Adaptive Viterbi Decoding for Modern Data Systems
Ziheng Deng, Xue Liu, Jiantong Jiang, Yankai Li, Qingxu Deng, Xiaochun Yang
Abstract
The Viterbi algorithm is a key operator for structured sequence inference in modern data systems, with applications in trajectory analysis, online recommendation, and speech recognition. As these workloads increasingly migrate to resource-constrained edge platforms, standard Viterbi decoding remains memory-intensive and computationally inflexible. Existing methods typically trade decoding time for space efficiency, but often incur significant runtime overhead and lack adaptability to various system constraints. This paper presents FLASH Viterbi, a Fast, Lightweight, Adaptive, and Hardware-Friendly Viterbi decoding operator that enhances adaptability and resource efficiency. FLASH Viterbi combines a non-recursive divide-and-conquer strategy with pruning and parallelization techniques to enhance both time and memory efficiency, making it well-suited for resource-constrained data systems. To further decouple space complexity from the hidden state space size, we present FLASH-BS Viterbi, a dynamic beam search variant built on a memory-efficient data structure. Both proposed algorithms exhibit strong adaptivity to diverse deployment scenarios by dynamically tuning internal parameters. To ensure practical deployment on edge devices, we also develop FPGA-based hardware accelerators for both algorithms, demonstrating high throughput and low resource usage. Extensive experiments show that our algorithms consistently outperform existing baselines in both decoding time and memory efficiency, while preserving adaptability and hardware-friendly characteristics essential for modern data systems. All codes are publicly available at https://github.com/Dzh-16/FLASH-Viterbi.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on5
- Local Path Integration for AttributionPeiyu Yang, Naveed Akhtar, Zeyi Wen, Ajmal MianAAAI 2023 · 16 citations
- LHMM: A Learning Enhanced HMM Model for Cellular Trajectory Map MatchingWeijie Shi, Jiajie Xu, Junhua Fang, Pingfu Chao et al.ICDE 2023 · 11 citations
- Indoor Mobility Semantics Annotation Using Coupled Conditional Markov NetworksHuan Li, Hua Lu, Muhammad Aamir Cheema, Lidan Shou et al.ICDE 2020 · 5 citations
- Fast Inference for Probabilistic Graphical ModelsJiantong Jiang, Zeyi Wen, Atif Bin Mansoor, Ajmal MianUSENIX ATC 2024 · 4 citations
- SIEVE: A Space-Efficient Algorithm for Viterbi DecodingMartino Ciaperoni, Aristides Gionis, Athanasios Katsamanis, Panagiotis KarrasSIGMOD 2022 · 3 citations
Related papers
- Faster Than Flash: Exploiting Attention Sparsity for Efficient Long-Context DecodingZhigeng Liu, Zhiyuan Ning, Ruixiao Li, Xiaoran Liu et al.ICML 2026
- Memory-Efficient KV Cache Optimization for Large Language Model Inference at the EdgeChi Zhang, Haisheng Tan, Haotian Pan, Yang Xu et al.INFOCOM 2026 · 1 citation
- Rethinking Pruning for Accelerating Deep Inference At the EdgeDawei Gao, Xiaoxi He, Zimu Zhou, Yongxin Tong et al.KDD 2020 · 24 citations
- InstAttention: In-Storage Attention Offloading for Cost-Effective Long-Context LLM InferenceXiurui Pan, Endian Li, Qiao Li, Shengwen Liang et al.HPCA 2025 · 22 citations
- VEDA: Efficient LLM Generation Through Voting-based KV Cache Eviction and Dataflow-flexible AcceleratorZhican Wang, Hongxiang Fan, Haroon Waris, Gang Wang et al.DAC 2025 · 1 citation
