Attention-assisted autoencoder neural network for end-to-end optimization of multi-access fiber-terahertz communication systems

We propose an end-to-end (E2E) fiber-terahertz (THz) integrated communication system based on an attention-assisted multi-access autoencoder (AMAE) neural network. The AMAE neural network comprises artificial neural networks (ANNs) that function as transmitters (T-ANNs), channel models, and receivers (R-ANNs) for multiple users. By connecting the computational graph of multiple T-ANNs and R-ANNs, we jointly optimize the AMAE to facilitate E2E multi-access communication. Attention mechanisms guide the optimization process to achieve fair and efficient power allocation and orthogonality among different users. We experimentally evaluated the performance of our proposed E2E framework in a 60 Gbit/s multi-channel (1, 5, and 10 km) fiber-THz hybrid system. The results indicate that our AMAE approach outperforms the conventional single-carrier quadrature amplitude modulation scheme by over 3 dB in receiver sensitivity and 11 Gbit/s in capacity under the 20% soft-decision forward error correction threshold in the same-channel back-to-back condition. Additionally, under the performance balance constraint, our approach achieves a transmission speed of 60 Gbit/s within a 10 GHz bandwidth in the multi-channel setting.

[1]  Junwen Zhang,et al.  Waveform-to-Waveform End-to-End Learning Framework in a Seamless Fiber-Terahertz Integrated Communication System , 2023, Journal of Lightwave Technology.

[2]  N. Chi,et al.  Deep-learning-based multi-user framework for end-to-end fiber-MMW communications. , 2023, Optics Express.

[3]  N. Chi,et al.  Two-dimensional End-to-end Deep Learning Autoencoder in G-Band Fiber-Terahertz Integrated Transmission for 6G RAN , 2022, 2022 27th OptoElectronics and Communications Conference (OECC) and 2022 International Conference on Photonics in Switching and Computing (PSC).

[4]  S. Calabrò,et al.  Nonlinear Equalization for Optical Communications Based on Entropy-Regularized Mean Square Error , 2022, 2022 European Conference on Optical Communication (ECOC).

[5]  Weisheng Hu,et al.  End-to-End Deep Learning for Long-haul Fiber Transmission Using Differentiable Surrogate Channel , 2022, Journal of Lightwave Technology.

[6]  B. Ommer,et al.  High-Resolution Image Synthesis with Latent Diffusion Models , 2021, 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[7]  Caijun Zhong,et al.  An Attention-Aided Deep Learning Framework for Massive MIMO Channel Estimation , 2021, IEEE Transactions on Wireless Communications.

[8]  Stefano Basagni,et al.  Intelligence and Learning in O-RAN for Data-Driven NextG Cellular Networks , 2020, IEEE Communications Magazine.

[9]  Xue Chen,et al.  Role of Digital Twin in Optical Communication: Fault Management, Hardware Configuration, and Transmission Simulation , 2020, ArXiv.

[10]  Gee-Kung Chang,et al.  Key Enabling Technologies for the Post-5G Era: Fully Adaptive, All-Spectra Coordinated Radio Access Network with Function Decoupling , 2020, IEEE Communications Magazine.

[11]  Fredrik Tufvesson,et al.  6G Wireless Systems: Vision, Requirements, Challenges, Insights, and Opportunities , 2020, Proceedings of the IEEE.

[12]  Yi Pan,et al.  Gradient Amplification: An efficient way to train deep neural networks , 2020, Big Data Min. Anal..

[13]  Laurent Schmalen,et al.  Concept and Experimental Demonstration of Optical IM/DD End-to-End System Optimization using a Generative Model , 2019, 2020 Optical Fiber Communications Conference and Exhibition (OFC).

[14]  Fayçal Ait Aoudia,et al.  Trainable Communication Systems: Concepts and Prototype , 2019, IEEE Transactions on Communications.

[15]  M. Z. Chowdhury,et al.  6G Wireless Communication Systems: Applications, Requirements, Technologies, Challenges, and Research Directions , 2019 .

[16]  Nan Chi,et al.  Two tributaries heterogeneous neural network based channel emulator for underwater visible light communication systems. , 2019, Optics express.

[17]  Jian Song,et al.  Joint Transceiver Optimization for Wireless Communication PHY Using Neural Network , 2019, IEEE Journal on Selected Areas in Communications.

[18]  Biing-Hwang Juang,et al.  Deep Learning-Based End-to-End Wireless Communication Systems With Conditional GANs as Unknown Channels , 2019, IEEE Transactions on Wireless Communications.

[19]  Paolo Torroni,et al.  Attention in Natural Language Processing , 2019, IEEE Transactions on Neural Networks and Learning Systems.

[20]  Polina Bayvel,et al.  End-to-End Optimized Transmission over Dispersive Intensity-Modulated Channels Using Bidirectional Recurrent Neural Networks , 2019, Optics express.

[21]  Lingchen Huang,et al.  AI Coding: Learning to Construct Error Correction Codes , 2019, IEEE Transactions on Communications.

[22]  Jakob Hoydis,et al.  Model-Free Training of End-to-End Communication Systems , 2018, IEEE Journal on Selected Areas in Communications.

[23]  Ami Wiesel,et al.  Learning to Detect , 2018, IEEE Transactions on Signal Processing.

[24]  Tobias A. Eriksson,et al.  Deep Learning of Geometric Constellation Shaping Including Fiber Nonlinearities , 2018, 2018 European Conference on Optical Communication (ECOC).

[25]  Polina Bayvel,et al.  End-to-End Deep Learning of Optical Fiber Communications , 2018, Journal of Lightwave Technology.

[26]  Zhang Qiang,et al.  Volterra and Wiener Equalizers for Short-Reach 100G PAM-4 Applications , 2017, Journal of Lightwave Technology.

[27]  Geoffrey Ye Li,et al.  Power of Deep Learning for Channel Estimation and Signal Detection in OFDM Systems , 2017, IEEE Wireless Communications Letters.

[28]  Stephan ten Brink,et al.  Deep Learning Based Communication Over the Air , 2017, IEEE Journal of Selected Topics in Signal Processing.

[29]  Lukasz Kaiser,et al.  Attention is All you Need , 2017, NIPS.

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

[31]  K. Zhong,et al.  Experimental demonstration of 608Gbit/s short reach transmission employing half-cycle 16QAM Nyquist-SCM signal and direct detection with 25Gbps EML. , 2016, Optics express.

[32]  Martín Abadi,et al.  TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems , 2016, ArXiv.

[33]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[34]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[35]  Yong Soo Cho,et al.  Adaptive precompensation of Wiener systems , 1998, IEEE Trans. Signal Process..

[36]  W. Marsden I and J , 2012 .