@inproceedings{
ida2024fast,
title={Fast Iterative Hard Thresholding Methods with Pruning Gradient Computations},
author={Yasutoshi Ida and Sekitoshi Kanai and Atsutoshi Kumagai and Tomoharu Iwata and Yasuhiro Fujiwara},
booktitle={The Thirty-eighth Annual Conference on Neural Information Processing Systems},
year={2024},
url={https://openreview.net/forum?id=09RKw0vXjR}
}
We accelerate the iterative hard thresholding (IHT) method, which finds \(k\) important elements from a parameter vector in a linear regression model. Although the plain IHT repeatedly updates the parameter vector during the optimization, computing gradients is the main bottleneck. Our method safely prunes unnecessary gradient computations to reduce the processing time.The main idea is to efficiently construct a candidate set, which contains \(k\) important elements in the parameter vector, for each iteration. Specifically, before computing the gradients, we prune unnecessary elements in the parameter vector for the candidate set by utilizing upper bounds on absolute values of the parameters. Our method guarantees the same optimization results as the plain IHT because our pruning is safe. Experiments show that our method is up to 73 times faster than the plain IHT without degrading accuracy.
Fast Iterative Hard Thresholding Methods with Pruning Gradient Computations
Yasutoshi Ida Sekitoshi Kanai Atsutoshi Kumagai Tomoharu Iwata Yasuhiro Fujiwara