Papers by Maryam Dialameh
FLOP-Efficient Training: Early Stopping Based on Test-Time Compute Awareness (2026.findings-acl)
Copied to clipboard
| Challenge: | Prior work shows that increasing test-time compute (TTC) can improve accuracy of large language models. |
| Approach: | They propose a TTC-aware training algorithm that jointly selects a checkpoint and a corresponding TTC configuration to minimize training compute without sacrificing accuracy. |
| Outcome: | The proposed method reduces training compute by 92% while maintaining accuracy. |
ECHO-LLaMA: Efficient Caching for High-Performance LLaMA Training (2025.emnlp-industry)
Copied to clipboard
Maryam Dialameh, Rezaul Karim, Hossein Rajabzadeh, Omar Mohamed Awad, Boxing Chen, Hyock Ju Kwon, Walid Ahmed, Yang Liu
| Challenge: | ECHO-LLaMA transforms LLa MA models into shared KV caching across certain layers, significantly reducing KV computational complexity while maintaining or improving language performance. |
| Approach: | They propose an efficient LLaMA architecture that transforms LLama models into shared KV caching across certain layers, reducing computational complexity while maintaining or improving language performance. |
| Outcome: | ECHO-LLaMA achieves up to 77% higher token-per-second throughput during training, up to 16% higher Model FLOPs Utilization (MFU) and up to 14% lower loss when trained on an equal number of tokens. |