acceptodds
Under review as a conference paper at ICLR 2027

MetaBrainPro: A Graph Learning Approach for Enterprise Metadata Lineage Retrieval

Abstract

We present MetaBrainPro, a Graphormer-style transformer architecture for enterprise metadata lineage retrieval. Our work addresses three practical challenges in scaling graph-former models: (1) gradient interference between prediction heads, (2) gradient vanishing under extreme topological penalty values, and (3) KV cache explosion during inference on large graphs. Our theoretical contributions are threefold: 1. Gradient Collusion Collapse Theorem: We prove that orthogonal gradients from discrete prediction heads collapse the representation norm to , explaining catastrophic performance degradation in multi-task settings. 2. Additive Spatial Bias Gradient Theorem: We demonstrate that additive bias operators preserve non-vanishing gradients even when attention scores saturate at extreme values (), providing provable gradient flow under topological pressure. 3. 1D TSI Channel Diversity Preservation Theorem: We prove that channel-wise adaptive scaling vectors maintain gradient diversity across all 512 dimensions, preventing representation collapse in long-term training. Our architecture integrates Additive Spatial Bias (eliminates gradient vanishing), MLA with KV compression to 128 dimensions (75% memory reduction), GQA with 4 groups, and Asymmetric Layered Gradient Clipping. The model achieves F1-Score of 0.9252 (Precision: 0.8757, Recall: 0.9806), significantly outperforming baseline GraphSAGE and standard Graphormer methods by approximately 37% absolute improvement in F1-Score. All theoretical guarantees are experimentally verified through ablation studies (M1-M6 variants). All results are averaged over 3 independent runs with different random seeds (42, 123, 456), reporting mean ± standard deviation. Statistical significance is assessed using two-tailed t-tests

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.