new

Get trending papers in your email inbox!

Subscribe

Daily Papers

byAK and the research community

Aug 20

Efficient Arbitrary Precision Acceleration for Large Language Models on GPU Tensor Cores

Large language models (LLMs) have been widely applied but face challenges in efficient inference. While quantization methods reduce computational demands, ultra-low bit quantization with arbitrary precision is hindered by limited GPU Tensor Core support and inefficient memory management, leading to suboptimal acceleration. To address these challenges, we propose a comprehensive acceleration scheme for arbitrary precision LLMs. At its core, we introduce a novel bipolar-INT data format that facilitates parallel computing and supports symmetric quantization, effectively reducing data redundancy. Building on this, we implement an arbitrary precision matrix multiplication scheme that decomposes and recovers matrices at the bit level, enabling flexible precision while maximizing GPU Tensor Core utilization. Furthermore, we develop an efficient matrix preprocessing method that optimizes data layout for subsequent computations. Finally, we design a data recovery-oriented memory management system that strategically utilizes fast shared memory, significantly enhancing kernel execution speed and minimizing memory access latency. Experimental results demonstrate our approach's effectiveness, with up to 2.4\times speedup in matrix multiplication compared to NVIDIA's CUTLASS. When integrated into LLMs, we achieve up to 6.7\times inference acceleration. These improvements significantly enhance LLM inference efficiency, enabling broader and more responsive applications of LLMs.

Proper motions of spectrally selected structures in the HH 83 outflow

We continue our program of investigation of the proper motions of spectrally separated structures in the Herbig-Haro outflows with the aid of Fabry-Perot scanning interferometry. This work mainly focuses on the physical nature of various structures in the jets. The aim of the present study is to measure the proper motions of the previously discovered kinematically separated structures in the working surface of the HH 83 collimated outflow. We used observations from two epochs separated by 15 years, which were performed on the 6m telescope with Fabry-Perot scanning interferometer. We obtained images corresponding to different radial velocities for the two separate epochs, and used them to measure proper motions. In the course of our data analysis, we discovered a counter bow-shock of HH 83 flow with positive radial velocity, which makes this flow a relatively symmetric bipolar system. The second epoch observations confirm that the working surface of the flow is split into two structures with an exceptionally large (250 km\ s^{-1}) difference in radial velocity. The proper motions of these structures are almost equal, which suggests that they are physically connected. The asymmetry of the bow shock and the turning of proper motion vectors suggests a collision between the outflow and a dense cloud. The profile of the Halpha line for the directly invisible infrared source HH 83 IRS, obtained by integration of the data within the reflection nebula, suggests it to be of P Cyg type with a broad absorption component characteristic of the FU Ori like objects. If this object underwent an FU Ori type outburst, which created the HH 83 working surfaces, its eruption took place about 1500 years ago according to the kinematical age of the outflow.

A Demographic-Conditioned Variational Autoencoder for fMRI Distribution Sampling and Removal of Confounds

Objective: fMRI and derived measures such as functional connectivity (FC) have been used to predict brain age, general fluid intelligence, psychiatric disease status, and preclinical neurodegenerative disease. However, it is not always clear that all demographic confounds, such as age, sex, and race, have been removed from fMRI data. Additionally, many fMRI datasets are restricted to authorized researchers, making dissemination of these valuable data sources challenging. Methods: We create a variational autoencoder (VAE)-based model, DemoVAE, to decorrelate fMRI features from demographics and generate high-quality synthetic fMRI data based on user-supplied demographics. We train and validate our model using two large, widely used datasets, the Philadelphia Neurodevelopmental Cohort (PNC) and Bipolar and Schizophrenia Network for Intermediate Phenotypes (BSNIP). Results: We find that DemoVAE recapitulates group differences in fMRI data while capturing the full breadth of individual variations. Significantly, we also find that most clinical and computerized battery fields that are correlated with fMRI data are not correlated with DemoVAE latents. An exception are several fields related to schizophrenia medication and symptom severity. Conclusion: Our model generates fMRI data that captures the full distribution of FC better than traditional VAE or GAN models. We also find that most prediction using fMRI data is dependent on correlation with, and prediction of, demographics. Significance: Our DemoVAE model allows for generation of high quality synthetic data conditioned on subject demographics as well as the removal of the confounding effects of demographics. We identify that FC-based prediction tasks are highly influenced by demographic confounds.