Deep Modulation Embedding

02/17/2019 ∙ by Amin Abbasloo, et al. ∙ 0

Deep neural network has recently shown very promising applications in different research directions and attracted the industry attention as well. Although the idea was introduced in the past but just recently the main limitation of using this class of algorithms is solved by enabling parallel computing on GPU hardware. Opening the possibility of hardware prototyping with proven superiority of this class of algorithm, trigger several research directions in communication system too. Among them cognitive radio, modulation recognition, learning based receiver and transceiver are already given very interesting result in simulation and real experimental evaluation implemented on software defined radio. Specifically, modulation recognition is mostly approached as a classification problem which is a supervised learning framework. But it is here addressed as an unsupervised problem with introducing new features for training, a new loss function and investigating the robustness of the pipeline against several mismatch conditions.



There are no comments yet.


page 3

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

I Introduction

Neural network was an active research area in the past and reached a point where no more investigation could be done because of its needs of computational resources for the training. Just recently, a few hardware companies released affordable GPU hardware that enables doing heavy computation in parallel. Consequently, API’s and compilers were developed and made public for handling the required computation for their training on GPU. This opens an opportunity for investigating the potentials of neural network even with more layers (deep network). Within few years, many different applications, new architectures and numerical techniques were suggested to address a wide range of tasks and fixing numerical issues of training the networks. Deep neural networks (DNN) showed incredible result outperforming the state-of-art algorithms and even pushed machine leaning techniques to a new level as well introduced new use cases that were possible only by human [1]

. Computer vision and graphics, speech and data science are the success stories of this newborn class of algorithms.

Electrical engineers have tested successfully the machine learning techniques for designing and analyzing the communication systems and signals. By re-introducing DNN, it was quite natural for communication system community to define projects for exploring the possibilities that this new paradigm can offer and even see the industrial applications

[2]. Moreover, NVIDIA released a new hardware family called ’Jetson’ that enables hardware prototyping specially for communication applications. The testbed is basically made of a software defined radio (SDR), Jetson GPU server and FPGA. SDR enables receiving and transmitting signals over channels and FPGA does the required computation real-time and fast. Jetson GPU server can hold the already trained network using a PC or laptop and later be stacked to the testbed hardware by a carrier for testing phase and real-world applications.

Among many research projects that have been launched, a few like cognitive radio, modulation recognition and learning based receiver, transceiver got the most attention although soon or later new applications will be out. The main difference between classical and DNN based method is, classical method are designed based on the expert knowledge but DNN based ones are basically data driven. The network learns during training the data space and can react accordingly during testing phase when it is fed by new/unseen data. Employing DNN in communication system includes improving the state-of-art design made by expert knowledge by replacing some blocks and modules by DNN. Moreover, one can also design the whole system based on DNN ignoring already established designs. Designing new DNN based receivers (keeping transmitter as a current state-of-art design) or a totally DNN based transceiver that both transmitter and receiver are replaced by DNN and trained over the channel are done/reported improving the capacity and reducing the bit error rate.

As the focus of this work is on modulation recognition, literature review mainly contains the already established and reported directions for cognitive radio and modulation recognition. Cognitive radio tries to monitor and manage the spectrum between users so it is required to detect the unused frequencies and identify the users (including the modulations). Detection is basically based by signal energy and modulation recognition is formulated by some expert knowledge and hand-crafted features like cyclostationary signatures in classical methods [3]. A few publications have addressed the modulation recognition as a classification problem. It is a supervised leaning that DNN learns the data space by knowing the ground truth as labels (each label shows a specific modulation). Using DNN for classification is hugely tested in image processing community and it also proved its performance in modulation recognition. Although there might be ambiguity for some modulations but a few extra steps for them can solve the issue [5, 6]

. Formulating the problem as an unsupervised learning is a flexible platform to deal with unlabelled data and provides a signature/feature for signal which can be used later for different tasks like classification, clustering and detection

[7, 8]. Since this perspective is not properly studied before and there are several open questions regarding the architecture, features and the robustness of the unsupervised techniques for modulation recognition. These give motivation for doing a systematic investigation along this line of work.

Section II explains the DNN architecture used for investigation and provides more details about loss function based on Markov modelling and new features for the training. Section III presents the training setup and discuss the experiments that have been done in order to see the effects of free parameters and robustness of the algorithm in mismatch conditions. A very cheap and affordable testbed has been introduced and implemented by HackRF One and GNURADIO.

Ii Network Architecture

Variational autoencoder is used as the network architecture for signal embedding. This is usually a natural choice for dimensionality reduction and has been used for denoising too [9]. The network is supposed to embed the input data with dimension in a new space with such that , here . The loss function can be defined in order to include more terms like Markov modelling [10]. During the optimization, the difference between the shifted signal and the decoder output with different lags is minimized, it has been inspired by Markov modelling. Features are I/Q components themselves, differences between I/Q components and their shifted versions (because the signal trajectory also shows some characteristics of the signal) and correlation with different lags.

Iii Result and Evaluation

Dataset is obtained from [4] and is generated using GNURADIO. Although, DNN usually needs GPU for training but the network architecture used here can be/was trained on CPU. The dimension of hidden layer was with depth equals to , learning decay and dropout. Since the network is pretty small and trains very fast even on CPU, it actually gives the opportunity to use it for primary evaluation with little budget. Its implementation is available on

Iii-a Offline

The network is trained on eleven different modulations with different SNR’s. For visualizing the result, I/Q trajectory is color coded with embedded space to show the signal characteristics and with a two-dimensional histogram of difference between embedded space and its first lag shift and embedded signal itself as a simple signature. After training, the network has been tested on unseen data for different SNR’s and modulations. Fig. 1 shows the result for a few seen SNR’s, each row corresponded to a specific SNR. Moreover, for example, (’WBFM’, ’10’) means the modulation is ’WBFM’ with SNR equals to 10 db. By decreasing the SNR, the signal characteristics changes gradually, Fig. 1 from top to bottom. Since lag differences and correlations are introduced as new features here, therefore the affect of increasing lag and including extra features are investigated too. In Fig. 2, first row shows the normalized distance between embedded spaces with lag equals to eight and sixteen. The second row is the distance between lag differences plus correlations with lag equals to eight and just lag differences with lag equals to sixteen as features (which both give more or less the same feature dimension). Increasing lags has impacts on all modulation but including correlation effects some more like ’PAM4’. Third row shows the distance if ’WBFM’ is included/excluded during the training. The network is able to create the signal characteristics even though it has not seen the modulation during the training phase. But it has a strong impact on other modulations as the concentration of dark color is increased for other modulations in the graphs. The forth row presents the distance if one specific SNR (6 db) is included/excluded during training (training data includes SNR’s up to 6 db, Fig. 1). Changing the conditions in general influences the modulations differently even for those with similar I/Q trajectories like ’AM-SSB’ and ’QAM16’. This shows that changing and including parameters, might help to embed visually similar signals but with different characteristics in a new space that is discriminating enough.

Iii-B Online

A GNURADIO data flow is set up to receive the data from a SDR and acts as a publish/subscribe server. The SDR hardware is a HackRf One which is very cheap and affordable and provides the possibility for developing a reasonable/low budget prototype for DNN based project in communication systems specially for student research projects. The recorded signal has been fed to the trained network. Fig3 shows the signal trajectory color coded with embedded space and two dimensional histogram as a simple signature. The station is a local FM radio station and its modulation is ’WBFM’. The histogram shows a flip along y-axis comparing to Fig. 1 for ’WBFM’. It is because of the rotation ambiguity of the embedded space.

Fig. 1: local FM radio signal with 102.4 MHz frequency in Bonn region. Left image shows signal trajectory color coded with embedded space and right is its two dimensional histogram.
Fig. 2: Five double-rows (one row shows the signal trajectory color coded by embedded space and the other is its two dimensional histogram of difference between embedded signal and its first lag shift and embedded signal itself) show signal trajectory for eleven different modulations (each row has eleven modulations and corresponds to a specific SNR). By decreasing the SNR, the two dimensional histogram, as a simple signal signature, slightly changes.
Fig. 3: Rows from top to bottom show the effect of adding more lags in the model, including lag difference with(out) correlation as features, removing one modulation from the training to see if network can predict it and the last row is the impact of SNR mismatch.

Iv Conclusion

Modulation recognition is addressed as an unsupervised learning problem here although it is shown in the literature that the performance of DNN based supervised learning is very good for classification. The embedded data later can be used for classification, clustering or detection. Although the autoencoder architecture is a very popular network for this task but for modulation recognition, it should be investigated what features and loss function can be added to the network in order to use the whole capacity of the architecture. Lag differences and correlations are introduced here for capturing the signal trajectory for a better embedding. Better embedding means similar signals end up close to and dissimilar far from each others. The loss function is quite easy to modify as it is done here based on Markov modelling. The effect of changing the lag and including extra features are discussed. Moreover, the effect of mismatch assuming two scenarios is studied. If a modulation and SNR were not part of training data, the network is robust enough to these cases and predict the unseen data. But embedded space is affected any how by all of them.

Appendix A Loss Function

The loss function is defined as Eq. 2 in [10] but the author also introduced another term for the loss function which did not decrease in the training for modulation embedding. However, any type of new terms can be easily added to the loss function to see the effect of it in the training.


This project is not supported by any official financial resources yet.