Risk Assessment Framework for Code LLMs via Leveraging Internal States
The pre-training paradigm plays a key role in the success of Large Language Models (LLMs), which have been recognized as one of the most significant advancements of AI recently. Building on these breakthroughs, code LLMs with advanced coding capabilities bring huge impacts on software engineering, showing the tendency to become an essential part of developers’ daily routines. However, the current code LLMs still face serious challenges related to trustworthiness, as they can generate incorrect, insecure, or unreliable code. Recent exploratory studies find that it can be promising to detect such risky outputs by analyzing LLMs’ internal states, akin to how the human brain unconsciously recognizes its own mistakes. Yet, most of these approaches are limited to narrow sub-domains of LLM operations and fall short of achieving industry-level scalability and practicability. To address these challenges, in this paper, we propose PtTrust, a two-stage risk assessment framework for code LLM based on internal state pre-training, designed to integrate seamlessly with the existing infrastructure of software companies. The core idea is that risk predictors could also undergo a pre-training process similar to LLMs. Specifically, PtTrust first performs unsupervised pre-training on large-scale unlabeled source code to learn general representations of LLM states. Then, it uses a small, labeled dataset to train a risk predictor. We demonstrate the effectiveness of PtTrust through fine-grained, code line-level risk assessment and demonstrate that it generalizes across tasks and different programming languages. Further experiments also reveal that PtTrust provides highly intuitive and interpretable features, fostering greater user trust. We believe PtTrust makes a promising step toward scalable and trustworthy assurance for code LLMs
Tue 24 JunDisplayed time zone: Amsterdam, Berlin, Bern, Rome, Stockholm, Vienna change
10:30 - 12:30 | SE for LLMJournal First / Industry Papers / Demonstrations / Research Papers / Ideas, Visions and Reflections at Cosmos 3C Chair(s): Hongyu Zhang Chongqing University | ||
10:30 10mTalk | Enhancing Code LLM Training with Programmer Attention Ideas, Visions and Reflections Yifan Zhang Vanderbilt University, Chen Huang Sichuan University, Zachary Karas Vanderbilt University, Thuy Dung Nguyen Vanderbilt University, Kevin Leach Vanderbilt University, Yu Huang Vanderbilt University | ||
10:40 20mTalk | Risk Assessment Framework for Code LLMs via Leveraging Internal States Industry Papers Yuheng Huang The University of Tokyo, Lei Ma The University of Tokyo & University of Alberta, Keizaburo Nishikino Fujitsu Limited, Takumi Akazaki Fujitsu Limited | ||
11:00 20mTalk | An Empirical Study of Issues in Large Language Model Training Systems Industry Papers Yanjie Gao Microsoft Research, Ruiming Lu Shanghai Jiao Tong University, Haoxiang Lin Microsoft Research, Yueguo Chen Renmin University of China DOI | ||
11:20 20mTalk | Look Before You Leap: An Exploratory Study of Uncertainty Analysis for Large Language Models Journal First Yuheng Huang The University of Tokyo, Jiayang Song University of Alberta, Zhijie Wang University of Alberta, Shengming Zhao University of Alberta, Huaming Chen The University of Sydney, Felix Juefei-Xu New York University, Lei Ma The University of Tokyo & University of Alberta Link to publication DOI Pre-print | ||
11:40 10mTalk | EvidenceBot: A Privacy-Preserving, Customizable RAG-Based Tool for Enhancing Large Language Model Interactions Demonstrations Nafiz Imtiaz Khan Department of Computer Science, University of California, Davis, Vladimir Filkov University of California at Davis, USA | ||
11:50 20mTalk | OpsEval: A Comprehensive Benchmark Suite for Evaluating Large Language Models’ Capability in IT Operations Domain Industry Papers Yuhe Liu Tsinghua University, Changhua Pei Computer Network Information Center at Chinese Academy of Sciences, Longlong Xu Tsinghua University, Bohan Chen Tsinghua University, Mingze Sun Tsinghua University, Zhirui Zhang Beijing University of Posts and Telecommunications, Yongqian Sun Nankai University, Shenglin Zhang Nankai University, Kun Wang Zhejiang University, Haiming Zhang Chinese Academy of Sciences, Jianhui Li Computer Network Information Center at Chinese Academy of Sciences, Gaogang Xie Computer Network Information Center at Chinese Academy of Sciences, Xidao Wen BizSeer, Xiaohui Nie Computer Network Information Center at Chinese Academy of Sciences, Minghua Ma Microsoft, Dan Pei Tsinghua University | ||
12:10 20mTalk | Hallucination Detection in Large Language Models with Metamorphic Relations Research Papers Borui Yang Beijing University of Posts ad Telecommunications, Md Afif Al Mamun University of Calgary, Jie M. Zhang King's College London, Gias Uddin York University, Canada DOI |
Cosmos 3C is the third room in the Cosmos 3 wing.
When facing the main Cosmos Hall, access to the Cosmos 3 wing is on the left, close to the stairs. The area is accessed through a large door with the number “3”, which will stay open during the event.