Neural Network Calibration for CTR Prediction

Ergun Biçici and Hasan Saribaş, Neural Network Calibration for CTR Prediction. 2023 8th International Conference on Computer Science and Engineering (UBMK), Burdur, Turkiye, 2023, pp. 473-476, doi: 10.1109/UBMK59864.2023.10286733. URL: https://ieeexplore.ieee.org/document/10286733

After a machine learning model has been trained, calibration techniques correct its prediction errors to produce predictions that are more robust and confident. As calibration techniques, we implement isotonic regression, Platt's scaling, neural networks, spline regression, and temperature scaling and test them on the prediction of click-through rate (CTR), which is an unbalanced task. We use 3 neural network based CTR prediction models on a publicly available CTR dataset and measure the improvements. Our findings show that isotonic regression is the fastest method whereas isotonic regression and spline regression are the two techniques that improves the performance the most.