← Search

Sheng Di

3 accepted papers

2026

Retrieval-of-Thought: Efficient Reasoning via Reusing Thoughts

ICLR 2026poster

Large reasoning models improve accuracy by producing long reasoning traces, but this inflates latency and cost, motivating inference-time efficiency. We propose Retrieval-of-Thought (RoT), which reuses prior reasoning as composable ``thought" steps to guide new problems. RoT organizes steps into a t…

Cited by 0SourcecodeScholar
2026

SPARe: Stacked Parallelism with Adaptive Reordering for Fault-Tolerant LLM Pretraining Systems with 100k+ GPUs

ICML 2026poster

In large-scale LLM pretraining systems with $100\mathrm{k}+$ GPUs, failures become the norm rather than the exception, and restart costs can dominate wall-clock training time. However, existing fault-tolerance mechanisms are largely unprepared for this restart-dominant regime. To address this challe…

Cited by 0SourceScholar
2024

FedFa: A Fully Asynchronous Training Paradigm for Federated Learning

IJCAI 2024poster

Federated learning has been identified as an efficient decentralized training paradigm for scaling the machine learning model training on a large number of devices while guaranteeing the data privacy of the trainers. FedAvg has become a foundational parameter update strategy for federated learning,…

Cited by 5SourcePDFScholar