arXiv subjects

Jinrong Jiang

Publications and source records attributed to Jinrong Jiang.

3 recordsLinked to original sources

arXiv2024

Unveiling and Controlling Anomalous Attention Distribution in Transformers

With the advent of large models based on the Transformer architecture, researchers have observed an anomalous phenomenon in the Attention mechanism--there is a very high attention on the first element, which is prevalent across Transformer-based models. It is crucial to understand it for the development of techniques focusing on attention distribution, such as Key-Value (KV) Cache compression and infinite extrapolation; however, the latent cause leaves to be unknown. In this paper, we analyze such a phenomenon from the perspective of waiver phenomenon, which involves reducing the internal values of certain elements in the sequence, allowing them to absorb excess attention without affecting their contribution to information. In specific models, due to differences in positional encoding and attention patterns, we have found that the selection of waiver elements by the model can be categorized into two methods: positional-encoding-based and feature-distribution-within-elements-based.

cs.LG↗

arXiv2022

CAINNFlow: Convolutional block Attention modules and Invertible Neural Networks Flow for anomaly detection and localization tasks

Detection of object anomalies is crucial in industrial processes, but unsupervised anomaly detection and localization is particularly important due to the difficulty of obtaining a large number of defective samples and the unpredictable types of anomalies in real life. Among the existing unsupervised anomaly detection and localization methods, the NF-based scheme has achieved better results. However, the two subnets (complex functions) $s_{i}(u_{i})$ and $t_{i}(u_{i})$ in NF are usually multilayer perceptrons, which need to squeeze the input visual features from 2D flattening to 1D, destroying the spatial location relationship in the feature map and losing the spatial structure information. In order to retain and effectively extract spatial structure information, we design in this study a complex function model with alternating CBAM embedded in a stacked $3\times3$ full convolution, which is able to retain and effectively extract spatial structure information in the normalized flow model. Extensive experimental results on the MVTec AD dataset show that CAINNFlow achieves advanced levels of accuracy and inference efficiency based on CNN and Transformer backbone networks as feature extractors, and CAINNFlow achieves a pixel-level AUC of $98.64\%$ for anomaly detection in MVTec AD.

cs.CV↗

arXiv2020

Portably parallel construction of a CI wave function from a matrix-product state using the Charm++ framework

The constructions of configuration interaction (CI) expansions from a matrix-product state (MPS) involves numerous matrix operations and the skillful sampling of important configurations when in a huge Hilbert space. In this work, we present an efficient procedure for constructing CI expansions from MPS using the Charm++ parallel programming framework, upon which automatic load balancing and object migration facilities can be employed. This procedure was employed in the MPS-to-CI utility (Moritz et al., J. Chem. Phys. 2007, 126, 224109), sampling-reconstructed complete active space algorithm (SR-CAS, Boguslawski et al., J. Chem. Phys. 2011, 134, 224101) and entanglement-driven genetic algorithm (EDGA, Luo et al., J. Chem. Theory Comput. 2017, 13, 4699-4710). It enhances productivity and allows the sampling programs evolve to their population-expansion versions (e.g., EDGA with population expansion [PE-EDGA]). Examples of 1,2-dioxetanone and firefly dioxetanone anion (FDO-) molecules demonstrated that 1) the procedure could be flexibly employed among various multi-core architectures; 2) the parallel efficiencies could be persistently improved simply by increasing the proportion of asynchronous executions; 3) PE-EDGA could construct a CAS-type CI wave function from a huge Hilbert space, with 0.9952 CI completeness and 96.7% correlation energy via ~1.66x10^6 configurations (only 0.0000028% of the total configurations) of a bi-radical state of FDO- molecule using the full valence active space within a few hours.

physics.comp-ph↗