Meta-ViterbiNet: Online Meta-Learned Viterbi Equalization for Non-Stationary Channels

Deep neural networks (DNNs) based digital receivers can potentially operate in complex environments. How-ever, the dynamic nature of communication channels implies that in some scenarios, DNN-based receivers should be periodically retrained in order to track temporal variations in the channel conditions. To this aim, frequent transmissions of lengthy pilot sequences are generally required, at the cost of substantial overhead. In this work we propose a DNN-aided symbol detector, Meta-ViterbiNet, that tracks channel variations with reduced overhead by integrating three complementary techniques: 1) We leverage domain knowledge to implement a model-based/data-driven equalizer, ViterbiNet, that operates with a relatively small number of trainable parameters; 2) We tailor a meta-learning procedure to the symbol detection problem, optimizing the hyperparameters of the learning algorithm to facilitate rapid online adaptation; and 3) We enable online training with short-length pilot blocks and coded data blocks. Numerical results demonstrate that Meta-ViterbiNet operates accurately in rapidly-varying channels, outperforming the previous best approach, based on ViterbiNet or conventional recurrent neural networks without meta-learning, by a margin of up to 0.6dB in bit error rate in various challenging scenarios. Index terms— Viterbi algorithm, meta-learning.

[1]  A. Lee Swindlehurst,et al.  Generative-Adversarial-Network Enabled Signal Detection for Communication Systems With Unknown Channel Models , 2021, IEEE Journal on Selected Areas in Communications.

[2]  Yonina C. Eldar,et al.  DeepSIC: Deep Soft Interference Cancellation for Multiuser MIMO Detection , 2020, IEEE Transactions on Wireless Communications.

[3]  Joonhyuk Kang,et al.  Learning to Demodulate From Few Pilots via Offline and Online Meta-Learning , 2019, IEEE Transactions on Signal Processing.

[4]  Yonina C. Eldar,et al.  Inference from Stationary Time Sequences via Learned Factor Graphs , 2020, ArXiv.

[5]  Yen-Liang Chen,et al.  Syndrome-Enabled Unsupervised Learning for Neural Network-Based Polar Decoder and Jointly Optimized Blind Equalizer , 2020, IEEE Journal on Emerging and Selected Topics in Circuits and Systems.

[6]  Dan Deng,et al.  A Note on Implementation Methodologies of Deep Learning-Based Signal Detection for Conventional MIMO Transmitters , 2020, IEEE Transactions on Broadcasting.

[7]  Joonhyuk Kang,et al.  End-to-End Fast Training of Communication Links Without a Channel Model via Online Meta-Learning , 2020, 2020 IEEE 21st International Workshop on Signal Processing Advances in Wireless Communications (SPAWC).

[8]  Yonina C. Eldar,et al.  Data-Driven Factor Graphs for Deep Symbol Detection , 2020, 2020 IEEE International Symposium on Information Theory (ISIT).

[9]  Yair Be’ery,et al.  Data-Driven Ensembles for Deep and Hard-Decision Hybrid Decoding , 2020, 2020 IEEE International Symposium on Information Theory (ISIT).

[10]  Joonhyuk Kang,et al.  From Learning to Meta-Learning: Reduced Training Overhead and Complexity for Communication Systems , 2020, 2020 2nd 6G Wireless Summit (6G SUMMIT).

[11]  Joonhyuk Kang,et al.  Meta-Learning to Communicate: Fast End-to-End Training for Fading Channels , 2019, ICASSP 2020 - 2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[12]  Yonina C. Eldar,et al.  ViterbiNet: A Deep Learning Based Viterbi Algorithm for Symbol Detection , 2019, IEEE Transactions on Wireless Communications.

[13]  Alexios Balatsoukas-Stimming,et al.  Deep Unfolding for Communications Systems: A Survey and Some New Directions , 2019, 2019 IEEE International Workshop on Signal Processing Systems (SiPS).

[14]  S. Brink,et al.  On Recurrent Neural Networks for Sequence-based Processing in Communications , 2019, 2019 53rd Asilomar Conference on Signals, Systems, and Computers.

[15]  Mihaela van der Schaar,et al.  Machine Learning in the Air , 2019, IEEE Journal on Selected Areas in Communications.

[16]  Himanshu Asnani,et al.  MIND: Model Independent Neural Decoder , 2019, 2019 IEEE 20th International Workshop on Signal Processing Advances in Wireless Communications (SPAWC).

[17]  Osvaldo Simeone,et al.  A Very Brief Introduction to Machine Learning With Applications to Communication Systems , 2018, IEEE Transactions on Cognitive Communications and Networking.

[18]  Andrea Goldsmith,et al.  Neural Network Detection of Data Sequences in Communication Systems , 2018, IEEE Transactions on Signal Processing.

[19]  Sergey Levine,et al.  Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks , 2017, ICML.

[20]  Jakob Hoydis,et al.  An Introduction to Deep Learning for the Physical Layer , 2017, IEEE Transactions on Cognitive Communications and Networking.

[21]  Claude Oestges,et al.  The COST 2100 MIMO channel model , 2011, IEEE Wirel. Commun..

[22]  John M. Cioffi,et al.  Iterative soft interference cancellation for multiple antenna systems , 2000, 2000 IEEE Wireless Communications and Networking Conference. Conference Record (Cat. No.00TH8540).

[23]  John Cocke,et al.  Optimal decoding of linear codes for minimizing symbol error rate (Corresp.) , 1974, IEEE Trans. Inf. Theory.

[24]  Andrew J. Viterbi,et al.  Error bounds for convolutional codes and an asymptotically optimum decoding algorithm , 1967, IEEE Trans. Inf. Theory.