arXiv subjects

Kai Zhang

Publications and source records attributed to Kai Zhang.

At least 307 records · Page 17Linked to original sources

arXiv2025

CHASE: A Native Relational Database for Hybrid Queries on Structured and Unstructured Data

Querying both structured and unstructured data has become a new paradigm in data analytics and recommendation. With unstructured data, such as text and videos, are converted to high-dimensional vectors and queried with approximate nearest neighbor search (ANNS). State-of-the-art database systems implement vector search as a plugin in the relational query engine, which tries to utilize the ANN index to enhance performance. After investigating a broad range of hybrid queries, we find that such designs may miss potential optimization opportunities and achieve suboptimal performance for certain queries. In this paper, we propose CHASE, a query engine that is natively designed to support efficient hybrid queries on structured and unstructured data. CHASE performs specific designs and optimizations on multiple stages in query processing. First, semantic analysis is performed to categorize queries and optimize query plans dynamically. Second, new physical operators are implemented to avoid redundant computations, which is the case with existing operators. Third, compilation-based techniques are adopted for efficient machine code generation. Extensive evaluations using real-world datasets demonstrate that CHASE achieves substantial performance improvements, with speedups ranging from 13% to an extraordinary 7500 times compared to existing systems. These results highlight CHASE's potential as a robust solution for executing hybrid queries.

cs.DB↗

arXiv2025

STAR: Spatial-Temporal Augmentation with Text-to-Video Models for Real-World Video Super-Resolution

Image diffusion models have been adapted for real-world video super-resolution to tackle over-smoothing issues in GAN-based methods. However, these models struggle to maintain temporal consistency, as they are trained on static images, limiting their ability to capture temporal dynamics effectively. Integrating text-to-video (T2V) models into video super-resolution for improved temporal modeling is straightforward. However, two key challenges remain: artifacts introduced by complex degradations in real-world scenarios, and compromised fidelity due to the strong generative capacity of powerful T2V models (\textit{e.g.}, CogVideoX-5B). To enhance the spatio-temporal quality of restored videos, we introduce\textbf{~\name} (\textbf{S}patial-\textbf{T}emporal \textbf{A}ugmentation with T2V models for \textbf{R}eal-world video super-resolution), a novel approach that leverages T2V models for real-world video super-resolution, achieving realistic spatial details and robust temporal consistency. Specifically, we introduce a Local Information Enhancement Module (LIEM) before the global attention block to enrich local details and mitigate degradation artifacts. Moreover, we propose a Dynamic Frequency (DF) Loss to reinforce fidelity, guiding the model to focus on different frequency components across diffusion steps. Extensive experiments demonstrate\textbf{~\name}~outperforms state-of-the-art methods on both synthetic and real-world datasets.

cs.CV↗

arXiv2024

RetrievalAttention: Accelerating Long-Context LLM Inference via Vector Retrieval

Transformer-based Large Language Models (LLMs) have become increasingly important. However, due to the quadratic time complexity of attention computation, scaling LLMs to longer contexts incurs extremely slow inference speed and high GPU memory consumption for caching key-value (KV) vectors. This paper proposes RetrievalAttention, a training-free approach to both accelerate attention computation and reduce GPU memory consumption. By leveraging the dynamic sparsity of attention mechanism, RetrievalAttention proposes to build approximate nearest neighbor search (ANNS) indexes for KV vectors in CPU memory and retrieve the most relevant ones through vector search during generation. Unfortunately, we observe that the off-the-shelf ANNS indexes are often ineffective for such retrieval tasks due to the out-of-distribution (OOD) between query vectors and key vectors in the attention mechanism. RetrievalAttention addresses the OOD challenge by designing an attention-aware vector search algorithm that can adapt to the distribution of query vectors. Our evaluation demonstrates that RetrievalAttention achieves near full attention accuracy while only requiring access to 1--3% of the data. This leads to a significant reduction in the inference cost of long-context LLMs, with a much lower GPU memory footprint. In particular, RetrievalAttention only needs a single NVIDIA RTX4090 (24GB) to serve 128K tokens for LLMs with 8B parameters, which is capable of generating one token in 0.188 seconds.

cs.LG↗

arXiv2024

AddSR: Accelerating Diffusion-based Blind Super-Resolution with Adversarial Diffusion Distillation

Blind super-resolution methods based on stable diffusion showcase formidable generative capabilities in reconstructing clear high-resolution images with intricate details from low-resolution inputs. However, their practical applicability is often hampered by poor efficiency, stemming from the requirement of thousands or hundreds of sampling steps. Inspired by the efficient adversarial diffusion distillation (ADD), we design~\name~to address this issue by incorporating the ideas of both distillation and ControlNet. Specifically, we first propose a prediction-based self-refinement strategy to provide high-frequency information in the student model output with marginal additional time cost. Furthermore, we refine the training process by employing HR images, rather than LR images, to regulate the teacher model, providing a more robust constraint for distillation. Second, we introduce a timestep-adaptive ADD to address the perception-distortion imbalance problem introduced by original ADD. Extensive experiments demonstrate our~\name~generates better restoration results, while achieving faster speed than previous SD-based state-of-the-art models (e.g., $7$$\times$ faster than SeeSR).

cs.CV↗

arXiv2024

Low-temperature mean valence of nickel ions in pressurized La$_3$Ni$_2$O$_7$

The discovery of high critical temperature (Tc) superconductivity in pressurized La$_3$Ni$_2$O$_7$ has ignited renewed excitement in the search of novel high-Tc superconducting compounds with 3d transition metals. Compared to other ambient-pressure superconductors, such as copper-oxide and iron-oxypnictides, unraveling the mechanisms of the pressure-induced superconductivity poses significant and unique challenges. A critical factor in this phenomenon seems to be related to the electronic configuration of 3d orbitals, which may play a fundamental role in driving high-Tc superconductivity. However, the pressure effects on the mixed-valence states of 3d-orbital cations and their influence on the emergence of high-Tc superconductivity remain poorly understood. Here, we use high-pressure (P) and low-temperature synchrotron X-ray absorption spectroscopy to investigate the influence of pressure on the mean valence change of Ni ions in La$_3$Ni$_2$O$_7$. Our results demonstrate that at a low-temperature of 20 K, the mean valence remains relatively stable across the pressures range from 1 atm to 40 GPa. Based on analyzing the absorption data, we find that, at a critical pressure, the ambient-pressure ordered phases disappear and both the structural and the superconducting phase transition occur. The pressure-induced structural phase transition revealed by our absorption results is consistent with that determined by X-ray diffraction, offering new information for a comprehensive understanding on the pressure-induced superconductivity in La$_3$Ni$_2$O$_7$.

cond-mat.supr-con↗

arXiv2024

CPath-Omni: A Unified Multimodal Foundation Model for Patch and Whole Slide Image Analysis in Computational Pathology

The emergence of large multimodal models (LMMs) has brought significant advancements to pathology. Previous research has primarily focused on separately training patch-level and whole-slide image (WSI)-level models, limiting the integration of learned knowledge across patches and WSIs, and resulting in redundant models. In this work, we introduce CPath-Omni, the first 15-billion-parameter LMM designed to unify both patch and WSI level image analysis, consolidating a variety of tasks at both levels, including classification, visual question answering, captioning, and visual referring prompting. Extensive experiments demonstrate that CPath-Omni achieves state-of-the-art (SOTA) performance across seven diverse tasks on 39 out of 42 datasets, outperforming or matching task-specific models trained for individual tasks. Additionally, we develop a specialized pathology CLIP-based visual processor for CPath-Omni, CPath-CLIP, which, for the first time, integrates different vision models and incorporates a large language model as a text encoder to build a more powerful CLIP model, which achieves SOTA performance on nine zero-shot and four few-shot datasets. Our findings highlight CPath-Omni's ability to unify diverse pathology tasks, demonstrating its potential to streamline and advance the field of foundation model in pathology.

cs.CV↗

arXiv2024

Effects of Thom disk on alleviating ground effects of a wall-mounted rotating cylinder

This study investigates the effects of Thom disks on alleviating ground effects by wall-mounted rotating cylinders, also known as Flettner rotors, which utilize wind energy for ship propulsion. Through three-dimensional direct numerical simulations, our findings reveal that introducing a secondary Thom disk near the ground significantly reduces the three-dimensional flow pattern induced by the ground, leading to a more uniform pressure distribution along the rotor's surface. We also explore how the vertical placement of a secondary Thom disk influences wake dynamics and aerodynamic forces. Optimal placement of the secondary disk is found at the ground, which maximizes the lift-to-drag ratio.

physics.flu-dyn↗

arXiv2024

RTCUDB: Building Databases with RT Processors

A spectrum of new hardware has been studied to accelerate database systems in the past decade. Specifically, CUDA cores are known to benefit from the fast development of GPUs and make notable performance improvements. The state-of-the-art GPU-based implementation, i.e., Crystal, can achieve up to 61 times higher performance than CPU-based implementations. However, experiments show that the approach has already saturated almost all GPU memory bandwidth, which means there is little room left for further performance improvements. We introduce RTCUDB, the first query engine that leverages ray tracing (RT) cores in GPUs to accelerate database query processing. RTCUDB efficiently transforms the evaluation of a query into a ray-tracing job in a three-dimensional space. By dramatically reducing the amount of accessed data and optimizing the data access pattern with the ray tracing mechanism, the performance of RTCUDB is no longer limited by the memory bandwidth as in CUDA-based implementations. Experimental results show that RTCUDB outperforms the state-of-the-art GPU-based query engine by up to 18.3 times while the memory bandwidth usage drops to only 36.7% on average.

cs.DB↗

arXiv2024

EP-CFG: Energy-Preserving Classifier-Free Guidance

Classifier-free guidance (CFG) is widely used in diffusion models but often introduces over-contrast and over-saturation artifacts at higher guidance strengths. We present EP-CFG (Energy-Preserving Classifier-Free Guidance), which addresses these issues by preserving the energy distribution of the conditional prediction during the guidance process. Our method simply rescales the energy of the guided output to match that of the conditional prediction at each denoising step, with an optional robust variant for improved artifact suppression. Through experiments, we show that EP-CFG maintains natural image quality and preserves details across guidance strengths while retaining CFG's semantic alignment benefits, all with minimal computational overhead.

cs.CV↗

arXiv2024

Diffusion Model with Representation Alignment for Protein Inverse Folding

Protein inverse folding is a fundamental problem in bioinformatics, aiming to recover the amino acid sequences from a given protein backbone structure. Despite the success of existing methods, they struggle to fully capture the intricate inter-residue relationships critical for accurate sequence prediction. We propose a novel method that leverages diffusion models with representation alignment (DMRA), which enhances diffusion-based inverse folding by (1) proposing a shared center that aggregates contextual information from the entire protein structure and selectively distributes it to each residue; and (2) aligning noisy hidden representations with clean semantic representations during the denoising process. This is achieved by predefined semantic representations for amino acid types and a representation alignment method that utilizes type embeddings as semantic feedback to normalize each residue. In experiments, we conduct extensive evaluations on the CATH4.2 dataset to demonstrate that DMRA outperforms leading methods, achieving state-of-the-art performance and exhibiting strong generalization capabilities on the TS50 and TS500 datasets.

cs.LG↗

arXiv2024

Accelerating Video Diffusion Models via Distribution Matching

Generative models, particularly diffusion models, have made significant success in data synthesis across various modalities, including images, videos, and 3D assets. However, current diffusion models are computationally intensive, often requiring numerous sampling steps that limit their practical application, especially in video generation. This work introduces a novel framework for diffusion distillation and distribution matching that dramatically reduces the number of inference steps while maintaining-and potentially improving-generation quality. Our approach focuses on distilling pre-trained diffusion models into a more efficient few-step generator, specifically targeting video generation. By leveraging a combination of video GAN loss and a novel 2D score distribution matching loss, we demonstrate the potential to generate high-quality video frames with substantially fewer sampling steps. To be specific, the proposed method incorporates a denoising GAN discriminator to distil from the real data and a pre-trained image diffusion model to enhance the frame quality and the prompt-following capabilities. Experimental results using AnimateDiff as the teacher model showcase the method's effectiveness, achieving superior performance in just four sampling steps compared to existing techniques.

cs.CV↗

arXiv2024

TTT-Unet: Enhancing U-Net with Test-Time Training Layers for Biomedical Image Segmentation

Biomedical image segmentation is crucial for accurately diagnosing and analyzing various diseases. However, Convolutional Neural Networks (CNNs) and Transformers, the most commonly used architectures for this task, struggle to effectively capture long-range dependencies due to the inherent locality of CNNs and the computational complexity of Transformers. To address this limitation, we introduce TTT-Unet, a novel framework that integrates Test-Time Training (TTT) layers into the traditional U-Net architecture for biomedical image segmentation. TTT-Unet dynamically adjusts model parameters during the testing time, enhancing the model's ability to capture both local and long-range features. We evaluate TTT-Unet on multiple medical imaging datasets, including 3D abdominal organ segmentation in CT and MR images, instrument segmentation in endoscopy images, and cell segmentation in microscopy images. The results demonstrate that TTT-Unet consistently outperforms state-of-the-art CNN-based and Transformer-based segmentation models across all tasks. The code is available at https://github.com/rongzhou7/TTT-Unet.

eess.IV↗

arXiv2024

Turbo3D: Ultra-fast Text-to-3D Generation

We present Turbo3D, an ultra-fast text-to-3D system capable of generating high-quality Gaussian splatting assets in under one second. Turbo3D employs a rapid 4-step, 4-view diffusion generator and an efficient feed-forward Gaussian reconstructor, both operating in latent space. The 4-step, 4-view generator is a student model distilled through a novel Dual-Teacher approach, which encourages the student to learn view consistency from a multi-view teacher and photo-realism from a single-view teacher. By shifting the Gaussian reconstructor's inputs from pixel space to latent space, we eliminate the extra image decoding time and halve the transformer sequence length for maximum efficiency. Our method demonstrates superior 3D generation results compared to previous baselines, while operating in a fraction of their runtime.

cs.CV↗

arXiv2024

Non-Bloch Dirac Points and Phase Diagram in the Stacked Non-Hermitian SSH Model

Topological semimetals exhibit protected band crossings in momentum space, accompanied by corresponding surface states. Non-Hermitian Hamiltonians introduce geometry-sensitive features that dissolve this bulk-boundary correspondence principle. In this paper, we exemplify this phenomenon by investigating a non-Hermitian 2D stacked SSH chain model with non-reciprocal hopping and on-site gain/loss. We derive an analytical phase diagram in terms of the complex energy gaps in the open-boundary spectrum. The phase diagram reveals the existence of non-Bloch Dirac points, which feature a real spectrum and only appear under open boundary conditions but disappear in Bloch bands under periodic boundary conditions. Due to the reality of the spectrum in the vicinity of non-Bloch Dirac points, we can locally map it to Hermitian semimetals within the Altland-Zirnabuer symmetry classes. Based on this mapping, we demonstrate that non-Bloch Dirac points are characterized by an integer topological charge. Unlike the band crossings in Hermitian semimetals, the locations of the non-Bloch Dirac points under different boundary geometries do not match each other, indicating a geometry-dependent bulk-boundary correspondence in non-Hermitian semimetals. Our findings provide new pathways into establishing unconventional bulk-boundary correspondence for non-Bloch Dirac metals in non-Hermitian systems.

cond-mat.mes-hall↗

arXiv2024

Magnetic twisting in an artificial ferrimagnet: Anisotropic magnetoresistance on Py/Gd/Py/Gd/Py/SiNx multilayers

The intensive study of non-collinear magnets promotes an urgent demand for the quantitative characterization of the non-collinear magnetic structures, which host numerous exotic phenomena. Here we systematically study the non-collinear magnetic structure of an artificial ferrimagnetic multilayer. The AMR measurements reveal two distinct twisted states whose magnetic structures can be quantitatively characterized with the assistance of micromagnetic simulations. Our results manifest AMR as an ideal probe of the non-collinear magnetic structure in artificial ferrimagnets.

cond-mat.mtrl-sci↗

arXiv2024

Buffer Anytime: Zero-Shot Video Depth and Normal from Image Priors

We present Buffer Anytime, a framework for estimation of depth and normal maps (which we call geometric buffers) from video that eliminates the need for paired video--depth and video--normal training data. Instead of relying on large-scale annotated video datasets, we demonstrate high-quality video buffer estimation by leveraging single-image priors with temporal consistency constraints. Our zero-shot training strategy combines state-of-the-art image estimation models based on optical flow smoothness through a hybrid loss function, implemented via a lightweight temporal attention architecture. Applied to leading image models like Depth Anything V2 and Marigold-E2E-FT, our approach significantly improves temporal consistency while maintaining accuracy. Experiments show that our method not only outperforms image-based approaches but also achieves results comparable to state-of-the-art video models trained on large-scale paired video datasets, despite using no such paired video data.

cs.CV↗

arXiv2024

Magnetochiral Charge Pumping due to Charge Trapping and Skin Effect in Chirality-Induced Spin Selectivity

Chirality-induced spin selectivity (CISS) generates giant spin polarization in transport through chiral molecules, paving the way for novel spintronic devices and enantiomer separation. Unlike conventional transport, CISS magnetoresistance (MR) violates Onsager's reciprocal relation, exhibiting significant resistance changes when reversing electrode magnetization at zero bias. However, its underlying mechanism remains unresolved. In this work, we propose that CISS MR originates from charge trapping that modifies the electron tunneling barrier and circumvents Onsager's relation, distinct from previous spin polarization-based models. Charge trapping is governed by the non-Hermitian skin effect, where dissipation leads to exponential wavefunction localization at the ferromagnet-chiral molecule interface. Reversing magnetization or chirality alters the localization direction, changing the occupation of impurity/defect states in the molecule (i.e., charge trapping) -- a phenomenon we term magnetochiral charge pumping. Our theory explains why CISS MR can far exceed the ferromagnet spin polarization and why chiral molecules violate the reciprocal relation but chiral metals do not. Furthermore, it predicts exotic phenomena beyond the conventional CISS framework, including asymmetric MR induced by magnetic fields alone (without ferromagnetic electrodes), as confirmed by recent experiments. This work offers a deeper understanding of CISS and opens avenues for controlling electrostatic interactions in chemical and biological systems through the magnetochiral charge pumping.

cond-mat.mes-hall↗

arXiv2024

Optimizing Student Ability Assessment: A Hierarchy Constraint-Aware Cognitive Diagnosis Framework for Educational Contexts

Cognitive diagnosis (CD) aims to reveal students' proficiency in specific knowledge concepts. With the increasing adoption of intelligent education applications, accurately assessing students' knowledge mastery has become an urgent challenge. Although existing cognitive diagnosis frameworks enhance diagnostic accuracy by analyzing students' explicit response records, they primarily focus on individual knowledge state, failing to adequately reflect the relative ability performance of students within hierarchies. To address this, we propose the Hierarchy Constraint-Aware Cognitive Diagnosis Framework (HCD), designed to more accurately represent student ability performance within real educational contexts. Specifically, the framework introduces a hierarchy mapping layer to identify students' levels. It then employs a hierarchy convolution-enhanced attention layer for in-depth analysis of knowledge concepts performance among students at the same level, uncovering nuanced differences. A hierarchy inter-sampling attention layer captures performance differences across hierarchies, offering a comprehensive understanding of the relationships among students' knowledge state. Finally, through personalized diagnostic enhancement, the framework integrates hierarchy constraint perception features with existing models, improving the representation of both individual and group characteristics. This approach enables precise inference of students' knowledge state. Research shows that this framework not only reasonably constrains changes in students' knowledge states to align with real educational settings, but also supports the scientific rigor and fairness of educational assessments, thereby advancing the field of cognitive diagnosis.

cs.CY↗