Demo Data from VLDB 2026 / reviewers in the wild / expert
Sixue Wang
dblp:369/7180
· DBLP profile ↗
1ranked-venue papers
0as first author
1since 2021 · last 2023
—ORCID · none
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 1 · 1 since 2021
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
1 paper |
Efficient and distributed learning · 67% Optimization for machine learning · 33% |
Topics — the 2 heaviest of 3, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Efficient and distributed learning › memory-efficient training
gradient checkpointing |
0.7 | 1 | 2023 | A fast heuristic to optimize time-space tradeoff for large models · NeurIPS 2023 |
Machine learning › Efficient and distributed learning
memory-efficient training |
0.7 | 1 | 2023 | A fast heuristic to optimize time-space tradeoff for large models · NeurIPS 2023 |
Methods — techniques the papers use, named apart from their topics
simulated annealing · 0.7mixed-integer linear programming · 0.7
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2023 | A fast heuristic to optimize time-space tradeoff for large modelsabstractTraining large-scale neural networks is heavily constrained by GPU memory. In order to circumvent this limitation, gradient checkpointing, or recomputation is a powerful technique. There is active research in this area with methods such as Checkmake or Moccasin. However, both Checkmate and Moccasin rely on mixed integer linear programming or constraint programming, resulting in limited scalability due to their exponentially large search space.
This paper proposes a novel algorithm for recomputation (FastSA) based on a simulated annealing heuristic that achieves comparable or even better solutions than state-of-the-art alternatives. FastSA can optimize computational graphs with thousands of nodes within 3 to 30 seconds, several orders of magnitude faster than current solutions.
We applied FastSA to PyTorch models and verified its effectiveness through popular large vision and text models, including recent language models with the transformer architecture. The results demonstrate significant memory reductions by 73% with extra 18% computational overheads on average. Our experiments demonstrate the practicality and effectiveness of our recomputation algorithm, further highlighting its potential for wide application in various deep learning domains. Akifumi Imanishi, Zijian Xu 0004, Masayuki Takagi, Sixue Wang, Emilio Castillo |
NeurIPS | 4 |