3 papers
cs.LG2025
Quality analysis and evaluation prediction of RAG retrieval based on machine learning algorithms
Ruoxin Zhang, Zhizhao Wen, Chao Wang +3
With the rapid evolution of large language models, retrieval enhanced generation technology has been widely used due to its ability to integrate external knowledge to improve outpu…
cs.LG2025
GPU Memory Requirement Prediction for Deep Learning Task Based on Bidirectional Gated Recurrent Unit Optimization Transformer
Chao Wang, Zhizhao Wen, Ruoxin Zhang +2
In response to the increasingly critical demand for accurate prediction of GPU memory resources in deep learning tasks, this paper deeply analyzes the current research status and i…
cs.LG2025
Optimization of bi-directional gated loop cell based on multi-head attention mechanism for SSD health state classification model
Zhizhao Wen, Ruoxin Zhang, Chao Wang
Aiming at the critical role of SSD health state prediction in data reliability assurance, this study proposes a hybrid BiGRU-MHA model that incorporates a multi-head attention mech…