Parameterized Abstract Interpretation for Transformer Verification
Pei Huang, Dennis Wei, Omri Isac, Haoze Wu, Min Wu, Clark W. Barrett
Abstract
Transformers based on the self-attention mechanism have become foundational models across a wide range of domains, thereby creating an urgent need for effective formal verification techniques to better understand their behavior and ensure safety guarantees. In this paper, we propose two parameterized linear abstract domains for the inner products in the self-attention module, aiming to improve verification precision. The first one constructs symbolic quadratic upper and lower bounds for the product of two scalars, and then derives parameterized affine bounds using tangents. The other one constructs parameterized bounds by interpolating affine bounds proposed in prior work. We evaluate these two parameterization methods and demonstrate that both of them outperform the state-of-the-art approach which is regarded as optimal with respect to a certain mean gap. Experimental results show that, in the context of robustness verification, our approach is able to verify many instances that cannot be verified by existing methods. In the interval analysis, our method achieves tighter results compared to the SOTA, with the strength becoming more pronounced as the network depth increases.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext cc47841f-e279-4a1c-8779-8aa0dde090ecCited by top-tier papers1
Ask how each one uses itBuilds on6
- Jailbroken: How Does LLM Safety Training Fail?Alexander Wei, Nika Haghtalab, Jacob SteinhardtNeurIPS 2023 · 2,230 citations
- AI2: Safety and Robustness Certification of Neural Networks with Abstract InterpretationTimon Gehr, Matthew Mirman, Dana Drachsler-Cohen, Petar Tsankov et al.S&P 2018 · 987 citations
- Robustness Verification for TransformersZhouxing Shi, Huan Zhang, Kai-Wei Chang, Minlie Huang et al.ICLR 2020 · 131 citations
- Scalable Neural Network Verification with Branch-and-bound Inferred Cutting PlanesDuo Zhou, Christopher Brix, Grani A. Hanasusanto, Huan ZhangNeurIPS 2024 · 49 citations
- Fast and precise certification of transformersGregory Bonaert, Dimitar I. Dimitrov, Maximilian Baader, Martin T. VechevPLDI 2021 · 18 citations
Related papers
- Precise Verification of Transformers Through ReLU-Catalyzed Abstraction RefinementHengjie Liu, Zhenya Zhang, Jianjun ZhaoCAV 2026
- Interval universal approximation for neural networksZi Wang, Aws Albarghouthi, Gautam Prakriya, Somesh JhaPOPL 2022 · 18 citations
- Input-Relational Verification of Deep Neural NetworksDebangshu Banerjee, Changming Xu, Gagandeep SinghPLDI 2024 · 9 citations
- A Tale of Two Approximations: Tightening Over-Approximation for DNN Robustness Verification via Under-ApproximationZhiyi Xue, Si Liu, Zhaodi Zhang, Yiting Wu et al.ISSTA 2023 · 5 citations
- Faith: An Efficient Framework for Transformer Verification on GPUsBoyuan Feng, Tianqi Tang, Yuke Wang, Zhaodong Chen et al.USENIX ATC 2022
