TPCNet : representation learning for H i mapping

We introduce TPCNet, a neural network predictor that combines convolutional and transformer architectures with positional encodings, for neutral atomic hydrogen (H i) spectral analysis. Trained on synthetic data sets, our models predict cold neutral gas fraction ($f_\mathrm{CNM}$) and H i opacity correction factor ($\mathcal {R_{\mathrm{H\,{\small I}}}$) from emission spectra based on the learned relationships between the desired output parameters and observables (optically thin column density and peak brightness). As a follow-up to Murray et al.’s shallow Convolutional Neural Network (CNN), we construct deep CNN models and compare them to TPCNet models. TPCNet outperforms deep CNNs, achieving a 10 per cent average increase in testing accuracy, algorithmic (training) stability, and convergence speed. Our findings highlight the robustness of the proposed model with sinusoidal positional encoding applied directly to the spectral input, addressing perturbations in training data set shuffling and convolutional network weight initializations. Higher spectral resolutions with increased spectral channels offer advantages, albeit with increased training time. Diverse synthetic data sets enhance model performance and generalization, as demonstrated by producing $f_\mathrm{CNM}$ and $\mathcal {R_{\mathrm{H\,{\small I}}}$ values consistent with evaluation ground truths. Applications of TPCNet to observed emission data reveal strong agreement between the predictions and Gaussian decomposition-based estimates (from emission and absorption surveys), emphasizing its potential in H i spectral analysis.