Memory-Efficient GPU Acceleration for Recommendation Systems with Large-Scale Top-K Retrieval

Authors

  • Ana Luisa Gaitan Department of Computer Science, George Mason University, Fairfax, VA, USA.
  • Germana Bernetti Department of Computer Science and Engineering, University at Buffalo, Buffalo, NY, USA.

Keywords:

recommender systems, GPU acceleration, top-K retrieval, memory efficiency, approximate nearest neighbor search, systems architecture, fairness, sustainability

Abstract

Large-scale recommender systems increasingly depend on accelerated retrieval pipelines to convert massive candidate sets into small ranked lists under strict latency and cost constraints. The central systems challenge is not merely scoring items but performing top-K retrieval over hundreds of millions or billions of embeddings while respecting limited GPU memory, bandwidth, and energy budgets. This paper examines memory-efficient GPU acceleration for recommendation systems with large-scale top-K retrieval. It argues that retrieval performance must be understood as a socio-technical property shaped by algorithmic choices, hardware architecture, deployment topology, governance policies, and sustainability commitments. The discussion first reviews the evolution from classical recommendation models to deep and retrieval-oriented architectures, then analyzes memory bottlenecks such as embedding tables, candidate indexes, intermediate score buffers, and kernel launch overheads. It explores compression, quantization, graph-based indexing, row-wise selection, multi-stage retrieval, and heterogeneous scheduling as mutually dependent design levers. The paper then connects these technical choices to infrastructure governance, fairness, robustness, and environmental accountability. It suggests that memory-efficient top-K acceleration is not a single optimization but a negotiated balance among recall, latency, fairness, reproducibility, and operational resilience. Evaluation practices must therefore include end-to-end system metrics, tail-latency behavior, workload skew, and drift-aware monitoring. The conclusion emphasizes that future recommendation platforms require co-designed retrieval kernels, memory hierarchies, policy controls, and sustainable deployment strategies rather than isolated GPU micro-optimizations.

References

1. Covington, P., Adams, J., & Sargin, E. (2016). Deep neural networks for YouTube recommendations. In Proceedings of the 10th ACM Conference on Recommender Systems (pp. 191-198). Association for Computing Machinery.

2. Naumov, M., Mudigere, D., Shi, H. J. M., Huang, J., Sundaraman, N., Park, J., Wang, X., Gupta, U., Wu, C. J., Azzolini, A. G., Dzhulgakov, D., Mallevich, A., Cherniavskii, I., Lu, Y., Krishnamoorthi, R., Yu, A., Kondratenko, V., Pereira, S., Chen, X., ... Smelyanskiy, M. (2019). Deep learning recommendation model for personalization and recommendation systems. arXiv preprint arXiv:1906.00091.

3. Cheng, H. T., Koc, L., Harmsen, J., Shaked, T., Chandra, T., Aradhye, H., Anderson, G., Corrado, G., Chai, W., Ispir, M., Anil, R., Haque, Z., Hong, L., Jain, V., Liu, X., & Shah, H. (2016). Wide and deep learning for recommender systems. In Proceedings of the 1st Workshop on Deep Learning for Recommender Systems (pp. 7-10). Association for Computing Machinery.

4. Rendle, S. (2010). Factorization machines. In 2010 IEEE International Conference on Data Mining (pp. 995-1000). IEEE.

5. Johnson, J., Douze, M., & Jégou, H. (2021). Billion-scale similarity search with GPUs. IEEE Transactions on Big Data, 7(3), 535-547.

6. Malkov, Y. A., & Yashunin, D. A. (2020). Efficient and robust approximate nearest neighbor search using hierarchical navigable small world graphs. IEEE Transactions on Pattern Analysis and Machine Intelligence, 42(4), 824-836.

7. Guo, R., Sun, P., Lindgren, E., Geng, Q., Simcha, D., Chern, F., & Kumar, S. (2020). Accelerating large-scale inference with anisotropic vector quantization. In Proceedings of the 37th International Conference on Machine Learning (pp. 3887-3896). PMLR.

8. Baranchuk, D., Persiyanov, D., Sinitsin, A., & Babenko, A. (2019). Learning to route in similarity graphs. In Proceedings of the 36th International Conference on Machine Learning (pp. 475-484). PMLR.

9. Chen, T., Li, M., Li, Y., Lin, M., Wang, N., Wang, M., Xiao, T., Xu, B., Zhang, C., & Zhang, Z. (2015). MXNet: A flexible and efficient machine learning library for heterogeneous distributed systems. arXiv preprint arXiv:1512.01274.

10. Jouppi, N. P., Young, C., Patil, N., Patterson, D., Agrawal, G., Bajwa, R., Bates, S., Bhatia, S., Boden, N., Borchers, A., Boyle, R., Cantin, P., Chao, C., Clark, C., Coriell, J., Daley, M., Dau, M., Dean, J., Gelb, B., ... Yoon, D. H. (2017). In-datacenter performance analysis of a tensor processing unit. In Proceedings of the 44th Annual International Symposium on Computer Architecture (pp. 1-12). Association for Computing Machinery.

11. Dean, J., & Barroso, L. A. (2013). The tail at scale. Communications of the ACM, 56(2), 74-80.

12. Abadi, M., Barham, P., Chen, J., Chen, Z., Davis, A., Dean, J., Devin, M., Ghemawat, S., Irving, G., Isard, M., Kudlur, M., Levenberg, J., Monga, R., Moore, S., Murray, D. G., Steiner, B., Tucker, P., Vasudevan, V., Warden, P., ... Zheng, X. (2016). TensorFlow: A system for large-scale machine learning. In 12th USENIX Symposium on Operating Systems Design and Implementation (pp. 265-283). USENIX Association.

13. Sculley, D., Holt, G., Golovin, D., Davydov, E., Phillips, T., Ebner, D., Chaudhary, V., Young, M., Crespo, J. F., & Dennison, D. (2015). Hidden technical debt in machine learning systems. In Advances in Neural Information Processing Systems 28 (pp. 2503-2511). Curran Associates.

14. Mehrabi, N., Morstatter, F., Saxena, N., Lerman, K., & Galstyan, A. (2021). A survey on bias and fairness in machine learning. ACM Computing Surveys, 54(6), 1-35.

15. Strubell, E., Ganesh, A., & McCallum, A. (2019). Energy and policy considerations for deep learning in NLP. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics (pp. 3645-3650). Association for Computational Linguistics.

16. Wu, C. J., Brooks, D., Chen, K., Chen, D., Choudhury, S., Dukhan, M., Hazelwood, K., Isaac, E., Jia, Y., Jia, B., Leyvand, T., Lu, H., Lu, Y., Qiao, L., Ranganathan, P., Shi, J., Wang, A., Wang, Y., Wei, W., ... Zhang, P. (2019). Machine learning at Facebook: Understanding inference at the edge. In 2019 IEEE International Symposium on High Performance Computer Architecture (pp. 331-344). IEEE.

17. Hazelwood, K., Bird, S., Brooks, D., Chintala, S., Diril, U., Dzhulgakov, D., Fawzy, M., Jia, B., Jia, Y., Kalro, A., Law, J., Lee, K., Lu, J., Noordhuis, P., Smelyanskiy, M., Xiong, L., & Wang, X. (2018). Applied machine learning at Facebook: A datacenter infrastructure perspective. In 2018 IEEE International Symposium on High Performance Computer Architecture (pp. 620-629). IEEE.

18. Xie, X., Luo, Y., Peng, H., & Ding, C. (2025, May). RTop-K: ultra-fast row-wise top-K selection for neural network acceleration on GPUs. In International Conference on Learning Representations (Vol. 2025, pp. 81200-81214).

Downloads

Published

2026-08-19