DTG-MA: Hard Guardrails Against Catastrophic Forgetting Through Attention Masking and Task Graphs
Authors/Creators
Description
DTG-MA (Dynamic Task-Graph Masked Attention) is a novel approach to solving catastrophic forgetting in continual learning. It provides hard architectural guarantees against forgetting by using attention masking and task graphs to isolate task-specific computation pathways in neural networks.
Key features:
- Hard isolation: Uses masking in attention mechanisms to physically prevent interference between task-specific pathways
- Task graph structure: Each task gets dedicated edges in the computation graph, preventing parameter reuse and conflicts
- Memory efficient: Frozen edges for completed tasks, eliminating the need for replay buffers or storing old model copies
- GPU-friendly: Standard tensor operations map cleanly to accelerators without exotic requirements
- Interpretability: Easier to understand which task-specific subgraph activated for a given prediction
Compared to alternatives like EWC, LwF, and Replay, DTG-MA trades potential transfer learning for predictable, guaranteed isolation when task boundaries are known. Ideal for domain models, multi-tenant systems, and scenarios where data cannot be replayed.
Files
dtgmav2.pdf
Files
(280.8 kB)
| Name | Size | Download all |
|---|---|---|
|
md5:f17674b09e8029c1e3184ff89c9f9fab
|
27.9 kB | Download |
|
md5:6d9d743ae5d790c5181077d010d9e109
|
253.0 kB | Preview Download |
Additional details
Related works
- Is supplement to
- Software documentation: https://github.com/infosave2007/dtgma (URL)
Dates
- Issued
-
2025-12-29
Software
- Repository URL
- https://github.com/infosave2007/dtgma
- Development Status
- Active
References
- Kirichenko, O. (2025). DTG-MA: Hard Guardrails Against Catastrophic Forgetting Through Attention Masking and Task Graphs. Zenodo. https://github.com/infosave2007/dtgma