LeoNet: A Machine Learning Method for Binary Pulsar Classification

Abstract

Binary pulsars provide valuable laboratories for testing theories of gravity, but orbital Doppler shifts complicate their detection. Fourier-domain acceleration and jerk searches address this challenge via matched filtering, but at substantial computational cost. We present LeoNet, a convolutional neural network that uses ten learnable filters to extract features of signals affected by Doppler shifts. The resulting ten-channel feature map provides a compact, lower-dimensional alternative to an explicitly sampled acceleration-jerk response grid and is analysed by a convolutional classifier to identify candidate signals. For simulated observations lasting 500 s, LeoNet achieves a mean relative reduction in false negative rate of 55.5% across five sampling intervals compared with the evaluated PRESTO acceleration-search configuration. TensorRT-optimised LeoNet processes each 500 s observation in 3.44-4.37 ms in FP32 on an NVIDIA H100 PCIe GPU across eight sampling intervals, including preprocessing, inference, and postprocessing. At a sampling interval of 128 microseconds, its mean processing time is 3.54 ms, compared with 1.767 s for PRESTO FDAS on an AMD EPYC 9825 CPU with search-frequency limits of 96-1000 Hz, corresponding to an approximately 499-fold speedup in the measured processing time. These results suggest that LeoNet has the potential to improve detection performance, while its millisecond-scale processing time supports its use as a candidate-identification stage in real-time binary pulsar search pipelines.