Adapting learning-based precoding across different system configurations is challenging due to multiple types of variables and constraints. While large-scale neural networks have been proposed for cross-task adaptation, whether such adaptability requires large models remains unclear. In this paper, we identify a structural property of a class of precoding problems: the subproblems associated with each type of variable in alternative optimization (AO) share a common computational structure across systems when other variables are fixed. This structural consistency enables the reuse of update rules across systems. Based on this observation, we propose a cross-system neural precoder (XNP), where each layer implements AO-inspired update equations, which define the layer-wise input-output mappings. By reusing common update structures and learning only lightweight nonlinear mappings, the XNP enables efficient adaptation across systems only with several thousand trainable parameters. Simulation results show that pre-trained XNPs achieve fast adaptation to new configurations with significantly fewer training samples and epochs than a graph neural network-based baseline. This demonstrates that cross-system adaptability can be achieved by exploiting shared computational structure, rather than relying on large models.
Graph neural networks (GNNs) have emerged as a promising approach to learning wireless policies efficiently by leveraging topology prior and incorporating relational inductive biases. However, when the optimal policy is not permutation equivariant (PE), conventional GNNs suffer from mismatched inductive biases, leading to degraded performance or poor generalizability. This issue arises in wireless tasks with expected objectives, such as channel estimation and end-to-end (E2E) precoding, where the PE property of the optimal policy depends on the underlying channel distribution. In this paper, we propose a novel positional attention-based GNN to learn permutation nonequivariant policies efficiently. The core idea is to incorporate relative positions of vertices into the attention mechanism via an embedding function, enabling the GNNs to capture asymmetric relationships. Consequently, the proposed GNN can represent permutation non-equivariant functions, while retaining high learning efficiency and size generalizability through parameter sharing. We consider channel estimation and E2E precoding as case studies, and prove that their policies are PE to users but not to antennas under spatially correlated channels. We employ the proposed GNN to learn the policies, where the embedding function is designed based on the channel covariance matrix. Simulation results demonstrate that the proposed GNN outperforms existing channel estimation and E2E precoding methods, requires fewer samples for training, and can be generalized to systems with different numbers of antennas and users.
Baichuan Zhao, Chenyang Yang, Jianyu Zhao et al.· 0 citations
We use cookies to run the site and, with your consent, for analytics and to show ads.
See our Cookie Policy.
