arXiv subjects

Ziyu Shu

Publications and source records attributed to Ziyu Shu.

6 recordsLinked to original sources

arXiv2025

Towards Low-Latency and Adaptive Ransomware Detection Using Contrastive Learning

Ransomware has become a critical threat to cybersecurity due to its rapid evolution, the necessity for early detection, and growing diversity, posing significant challenges to traditional detection methods. While AI-based approaches had been proposed by prior works to assist ransomware detection, existing methods suffer from three major limitations, ad-hoc feature dependencies, delayed response, and limited adaptability to unseen variants. In this paper, we propose a framework that integrates self-supervised contrastive learning with neural architecture search (NAS) to address these challenges. Specifically, this paper offers three important contributions. (1) We design a contrastive learning framework that incorporates hardware performance counters (HPC) to analyze the runtime behavior of target ransomware. (2) We introduce a customized loss function that encourages early-stage detection of malicious activity, and significantly reduces the detection latency. (3) We deploy a neural architecture search (NAS) framework to automatically construct adaptive model architectures, allowing the detector to flexibly align with unseen ransomware variants. Experimental results show that our proposed method achieves significant improvements in both detection accuracy (up to 16.1%) and response time (up to 6x) compared to existing approaches while maintaining robustness under evasive attacks.

cs.CR↗

arXiv2025

SAND: A Self-supervised and Adaptive NAS-Driven Framework for Hardware Trojan Detection

The globalized semiconductor supply chain has made Hardware Trojans (HT) a significant security threat to embedded systems, necessitating the design of efficient and adaptable detection mechanisms. Despite promising machine learning-based HT detection techniques in the literature, they suffer from ad hoc feature selection and the lack of adaptivity, all of which hinder their effectiveness across diverse HT attacks. In this paper, we propose SAND, a selfsupervised and adaptive NAS-driven framework for efficient HT detection. Specifically, this paper makes three key contributions. (1) We leverage self-supervised learning (SSL) to enable automated feature extraction, eliminating the dependency on manually engineered features. (2) SAND integrates neural architecture search (NAS) to dynamically optimize the downstream classifier, allowing for seamless adaptation to unseen benchmarks with minimal fine-tuning. (3) Experimental results show that SAND achieves a significant improvement in detection accuracy (up to 18.3%) over state-of-the-art methods, exhibits high resilience against evasive Trojans, and demonstrates strong generalization.

cs.CR↗

arXiv2024

Residual Back Projection With Untrained Neural Networks

Background and Objective: The success of neural networks in a number of image processing tasks has motivated their application in image reconstruction problems in computed tomography (CT). While progress has been made in this area, the lack of stability and theoretical guarantees for accuracy, together with the scarcity of high-quality training data for specific imaging domains pose challenges for many CT applications. In this paper, we present a framework for iterative reconstruction (IR) in CT that leverages the hierarchical structure of neural networks, without the need for training. Our framework incorporates this structural information as a deep image prior (DIP), and uses a novel residual back projection (RBP) connection that forms the basis for our iterations. Methods: We propose using an untrained U-net in conjunction with a novel residual back projection to minimize an objective function and achieve high-accuracy reconstruction. In each iteration, the weights of the untrained U-net are optimized, and the output of the U-net in the current iteration is used to update the input of the U-net in the next iteration through the aforementioned RBP connection. Results: Experimental results demonstrate that the RBP-DIP framework offers improvements over other state-of-the-art conventional IR methods, as well as pre-trained and untrained models with similar network structures under multiple conditions. These improvements are particularly significant in the few-view, limited-angle, and low-dose imaging configurations. Conclusions: Applying to both parallel and fan beam X-ray imaging, our framework shows significant improvement under multiple conditions. Furthermore, the proposed framework requires no training data and can be adjusted on-demand to adapt to different conditions (e.g. noise level, geometry, and imaged object).

eess.IV↗

arXiv2024

SDIP: Self-Reinforcement Deep Image Prior Framework for Image Processing

Deep image prior (DIP) proposed in recent research has revealed the inherent trait of convolutional neural networks (CNN) for capturing substantial low-level image statistics priors. This framework efficiently addresses the inverse problems in image processing and has induced extensive applications in various domains. However, as the whole algorithm is initialized randomly, the DIP algorithm often lacks stability. Thus, this method still has space for further improvement. In this paper, we propose the self-reinforcement deep image prior (SDIP) as an improved version of the original DIP. We observed that the changes in the DIP networks' input and output are highly correlated during each iteration. SDIP efficiently utilizes this trait in a reinforcement learning manner, where the current iteration's output is utilized by a steering algorithm to update the network input for the next iteration, guiding the algorithm toward improved results. Experimental results across multiple applications demonstrate that our proposed SDIP framework offers improvement compared to the original DIP method and other state-of-the-art methods.

cs.CV↗

arXiv2022

A Global Constraint to Improve CT Reconstruction Under Non-Ideal Conditions

Background and Objective: The strong demand for medical imaging applications leads to the popularity of the CT reconstruction problem. Researchers proposed multiple constraints to tackle none ideal factors in CT reconstruction such as sparse-view, limited-angle, and low-dose conditions. Most of these constraints such as total variation are local constraints focusing on the relationship between a pixel and its neighbors. In this paper, we propose a new constraint utilizing the global prior of CT images to greatly reduce the streak artifacts and further improve the reconstruction accuracy. Methods: A CT image of the human body contains a limited number of different types of tissues, so pixels in CT images can be grouped into several groups according to their corresponding types. In our work, we focus on the composition classification for individual pixels and utilize it as a global prior, which differs from priors utilized by most current constraints. We propose segmenting pixels based on their gray levels during the reconstruction process, and forcing pixels in the same group to have similar gray levels. Results: Our experiments on the Shepp-Logan phantom and two real CT images from different benchmarks show that the proposed constraint can help the conventional local constraints further improve the reconstruction results under sparse-view, limited-angle, and low-dose conditions. Conclusions: Different from most current constraints focusing on the local prior, our proposed constraint only utilizes the global prior of CT images. In that case, our proposed constraint can collaborate with most local constraints and improve the reconstruction quality significantly. Furthermore, the proposed constraint also has the potential for further improvement, as the composition classification can be done with some more delicate methods, such as neural network related semantic segmentation algorithms.

eess.IV↗

arXiv2020

Gram filtering and sinogram interpolation for pixel-basis in parallel-beam X-ray CT reconstruction

The key aspect of parallel-beam X-ray CT is forward and back projection, but its computational burden continues to be an obstacle for applications. We propose a method to improve the performance of related algorithms by calculating the Gram filter exactly and interpolating the sinogram signal optimally. In addition, the detector blur effect can be included in our model efficiently. The improvements in speed and quality for back projection and iterative reconstruction are shown in our experiments on both analytical phantoms and real CT images.

eess.IV↗