CDiT: Conditional Diffusion Transformer for Geometry-Aware Terahertz Cross Far- and Near-Field Channel Generation

Abstract: Accurate channel modeling is fundamental to design and evaluation of Terahertz (THz) ultra-massive multiple-input multiple-output (UM-MIMO) systems. However, existing model-based approaches typically rely on simplified assumptions, such as sparsity or predefined parametric structures, which are insufficient to capture the complex spatial variations and cross far-/near-field propagation characteristics of practical THz channels. In this paper, a conditional diffusion transformer (CDiT) framework is proposed for high-fidelity THz channel generation. By leveraging the state-of-the-art hybrid planar-spherical wave model (HPSM), THz channel modeling is formulated as a geometry-aware conditional generative learning problem in the sparse beamspace domain. Position information is incorporated as a conditioning signal within a diffusion-transformer architecture, enabling effective learning of the spatially dependent channel distribution. By combining the strong distribution modeling capability of diffusion models with the global dependency modeling strength of transformers, the proposed framework achieves controllable and high-fidelity THz channel synthesis. Extensive experiments on realistic THz channel datasets demonstrate that the proposed framework converges stably and significantly outperforms representative benchmark methods. The proposed framework provides a promising data-driven paradigm for THz channel modeling in next-generation wireless systems.