Extending FISTA to FISTA-Net: Adaptive reflection parameters fitting for the deconvolution-based sound source localization in the reverberation environment

Noise source localization is considered a key technology for the low-noise design of machinery and equipment. The spatial resolution of conventional beamforming localization methods is effectively improved by the fast iterative shrinkage thresholding algorithm(FISTA). The application of FISTA for sound source localization in reverberant environments is limited by the selection of reflection parameters. In this paper, FISTA extends to FISTA-Net and is proposed to solve the problem of adaptive fitting of reflection parameters to improve the performance of sound source localization in a reverberant environment. In FISTA-Net, the network model can be established by expanding and truncating the FISTA algorithm based on a fixed number of iterations. The parameter iteration steps, shrinkage thresholds and update operators are updated in the FISTA-Net model by training. A series of simulations are conducted for various frequencies, reverberation times, and signal-to-noise ratios and compared to several de-reverberation algorithms to evaluate the performance of the proposed algorithm. The loudspeaker source localization experiment was conducted in a closed room to verify the effectiveness of FISTA-Net in real-world scenarios.