2 papers
cs.PF2025
xMem: A CPU-Based Approach for Accurate Estimation of GPU Memory in Deep Learning Training Workloads
Jiabo Shi, Dimitrios Pezaros, Yehia Elkhatib
The global scarcity of GPUs necessitates more sophisticated strategies for Deep Learning jobs in shared cluster environments. Accurate estimation of how much GPU memory a job will…
cs.LG2024
Personalizing Low-Rank Bayesian Neural Networks Via Federated Learning
Boning Zhang, Dongzhu Liu, Osvaldo Simeone +3
To support real-world decision-making, it is crucial for models to be well-calibrated, i.e., to assign reliable confidence estimates to their predictions. Uncertainty quantificatio…