paper-with-me

홈 › Papers

On Trojan Signatures in Large Language Models of Code

2024-02-23 · Aftab Hussain, Md Rafiqul Islam Rabin, Mohammad Amin Alipour

Trojan signatures, as described by Fields et al. (2021), are noticeable differences in the distribution of the trojaned class parameters (weights) and the non-trojaned class parameters of the trojaned model, that can be used to detect the trojaned model. Fields et al. (2021) found trojan signatures in computer vision classification tasks with image models, such as, Resnet, WideResnet, Densenet, and VGG. In this paper, we investigate such signatures in the classifier layer parameters of large language models of source code. Our results suggest that trojan signatures could not generalize to LLMs of code. We found that trojaned code models are stubborn, even when the models were poisoned under more explicit settings (finetuned with pre-trained weights frozen). We analyzed nine trojaned models for two binary classification tasks: clone and defect detection. To the best of our knowledge, this is the first work to examine weight-based trojan signature revelation techniques for large-language models of code and furthermore to demonstrate that detecting trojans only from the weights in such models is a hard problem.

📄 PDF Abstract BibTeX arXiv:2402.16896

Code (0)

등록된 구현이 없습니다.

Tasks

Binary ClassificationDefect Detection

Methods 이 논문이 사용한 방법론

Dropout Dropout is a regularization technique for neural networks that drops a unit (along with connections) at training time with a specified probability $p$ (a common value is…
Convolution A convolution is a type of matrix operation, consisting of a kernel, a small matrix of weights, that slides over input data performing element-wise multiplication with the…
Dense Connections Dense Connections, or Fully Connected Connections, are a type of layer in a deep neural network that use a linear operation where every input is connected to every output…
Max Pooling Max Pooling is a pooling operation that calculates the maximum value for patches of a feature map, and uses it to create a downsampled (pooled) feature map. It is usually…
Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…

Similar Papers 제목 키워드 기반

TRIGS: Trojan Identification from Gradient-based Signatures

2023-06-08 · Mohamed E. Hussein, Sudharshan Subramaniam Janakiraman, Wael AbdAlmageed

Training machine learning models can be very expensive or even unaffordable. This may be, for example, due to data limitations, such as unavailability or being too large, or computational power limitations. Therefore, it…

Unlearning Trojans in Large Language Models: A Comparison Between Natural Language and Source Code

2024-08-22 · Mahdi Kazemi, Aftab Hussain, Md Rafiqul Islam Rabin, Mohammad Amin Alipour 외

This work investigates the application of Machine Unlearning (MU) for mitigating the impact of trojans embedded in conventional large language models of natural language (Text-LLMs) and large language models of code (Cod…

Defect DetectionMachine UnlearningSentiment Analysis

Trojan Signatures in DNN Weights

2021-09-07 · Greg Fields, Mohammad Samragh, Mojan Javaheripi, Farinaz Koushanfar 외

Deep neural networks have been shown to be vulnerable to backdoor, or trojan, attacks where an adversary has embedded a trigger in the network at training time such that the model correctly classifies all standard inputs…

Trojans in Large Language Models of Code: A Critical Review through a Trigger-Based Taxonomy

2024-05-05 · Aftab Hussain, Md Rafiqul Islam Rabin, Toufique Ahmed, Bowen Xu 외

Large language models (LLMs) have provided a lot of exciting new capabilities in software development. However, the opaque nature of these models makes them difficult to reason about and inspect. Their opacity gives rise…

TrojanDec: Data-free Detection of Trojan Inputs in Self-supervised Learning

2025-01-07 · Yupei Liu, Yanting Wang, Jinyuan Jia

An image encoder pre-trained by self-supervised learning can be used as a general-purpose feature extractor to build downstream classifiers for various downstream tasks. However, many studies showed that an attacker can …

Self-Supervised Learning