arXiv subjects

Lei Shi

Publications and source records attributed to Lei Shi.

At least 109 records · Page 6Linked to original sources

arXiv2025

Leveraging semantic similarity for experimentation with AI-generated treatments

Large Language Models (LLMs) enable a new form of digital experimentation where treatments combine human and model-generated content in increasingly sophisticated ways. The main methodological challenge in this setting is representing these high-dimensional treatments without losing their semantic meaning or rendering analysis intractable. Here, we address this problem by focusing on learning low-dimensional representations that capture the underlying structure of such treatments. These representations enable downstream applications such as guiding generative models to produce meaningful treatment variants and facilitating adaptive assignment in online experiments. We propose double kernel representation learning, which models the causal effect through the inner product of kernel-based representations of treatments and user covariates. We develop an alternating-minimization algorithm that learns these representations efficiently from data and provides convergence guarantees under a low-rank factor model. As an application of this framework, we introduce an adaptive design strategy for online experimentation and demonstrate the method's effectiveness through numerical experiments.

stat.ME↗

arXiv2025

Uncertainty-Aware Extreme Point Tracing for Weakly Supervised Ultrasound Image Segmentation

Automatic medical image segmentation is a fundamental step in computer-aided diagnosis, yet fully supervised approaches demand extensive pixel-level annotations that are costly and time-consuming. To alleviate this burden, we propose a weakly supervised segmentation framework that leverages only four extreme points as annotation. Specifically, bounding boxes derived from the extreme points are used as prompts for the Segment Anything Model 2 (SAM2) to generate reliable initial pseudo labels. These pseudo labels are progressively refined by an enhanced Feature-Guided Extreme Point Masking (FGEPM) algorithm, which incorporates Monte Carlo dropout-based uncertainty estimation to construct a unified gradient uncertainty cost map for boundary tracing. Furthermore, a dual-branch Uncertainty-aware Scale Consistency (USC) loss and a box alignment loss are introduced to ensure spatial consistency and precise boundary alignment during training. Extensive experiments on two public ultrasound datasets, BUSI and UNS, demonstrate that our method achieves performance comparable to, and even surpassing fully supervised counterparts while significantly reducing annotation cost. These results validate the effectiveness and practicality of the proposed weakly supervised framework for ultrasound image segmentation.

cs.CV↗

arXiv2025

Revisit First-order Methods for Geodesically Convex Optimization

In a seminal work of Zhang and Sra, gradient descent methods for geodesically convex optimization were comprehensively studied. In particular, Zhang and Sra derived a comparison inequality that relates the iterative points in the optimization process. Since their seminal work, numerous follow-ups have studied different downstream usages of their comparison lemma. In this work, we introduce the concept of quasilinearization to optimization, presenting a novel framework for analyzing geodesically convex optimization. By leveraging this technique, we establish state-of-the-art convergence rates -- for both deterministic and stochastic settings -- under weaker assumptions than previously required. The technique of quasilinearization may prove valuable for other non-Euclidean optimization problems.

math.OC↗

arXiv2025

Expressive Power of Deep Networks on Manifolds: Simultaneous Approximation

A key challenge in scientific machine learning is solving partial differential equations (PDEs) on complex domains, where the curved geometry complicates the approximation of functions and their derivatives required by differential operators. This paper establishes the first simultaneous approximation theory for deep neural networks on manifolds. We prove that a constant-depth $\mathrm{ReLU}^{k-1}$ network with bounded weights--a property that plays a crucial role in controlling generalization error--can approximate any function in the Sobolev space $\mathcal{W}_p^{k}(\mathcal{M}^d)$ to an error of $\varepsilon$ in the $\mathcal{W}_p^{s}(\mathcal{M}^d)$ norm, for $k\geq 3$ and $s<k$, using $\mathcal{O}(\varepsilon^{-d/(k-s)})$ nonzero parameters, a rate that overcomes the curse of dimensionality by depending only on the intrinsic dimension $d$. These results readily extend to functions in Hölder-Zygmund spaces. We complement this result with a matching lower bound, proving our construction is nearly optimal by showing the required number of parameters matches up to a logarithmic factor. Our proof of the lower bound introduces novel estimates for the Vapnik-Chervonenkis dimension and pseudo-dimension of the network's high-order derivative classes. These complexity bounds provide a theoretical cornerstone for learning PDEs on manifolds involving derivatives. Our analysis reveals that the network architecture leverages a sparse structure to efficiently exploit the manifold's low-dimensional geometry. Finally, we corroborate our theoretical findings with numerical experiments.

math.NA↗

arXiv2025

ALLabel: Three-stage Active Learning for LLM-based Entity Recognition using Demonstration Retrieval

Many contemporary data-driven research efforts in the natural sciences, such as chemistry and materials science, require large-scale, high-performance entity recognition from scientific datasets. Large language models (LLMs) have increasingly been adopted to solve the entity recognition task, with the same trend being observed on all-spectrum NLP tasks. The prevailing entity recognition LLMs rely on fine-tuned technology, yet the fine-tuning process often incurs significant cost. To achieve a best performance-cost trade-off, we propose ALLabel, a three-stage framework designed to select the most informative and representative samples in preparing the demonstrations for LLM modeling. The annotated examples are used to construct a ground-truth retrieval corpus for LLM in-context learning. By sequentially employing three distinct active learning strategies, ALLabel consistently outperforms all baselines under the same annotation budget across three specialized domain datasets. Experimental results also demonstrate that selectively annotating only 5\%-10\% of the dataset with ALLabel can achieve performance comparable to the method annotating the entire dataset. Further analyses and ablation studies verify the effectiveness and generalizability of our proposal.

cs.CL↗

arXiv2025

Signal Lasso with Non-Convex Penalties for Efficient Network Reconstruction and Topology Inference

Inferring network structures remains an interesting question for its importance on the understanding and controlling collective dynamics of complex systems. The existing shrinking methods such as Lasso-type estimation can not suitably reveal such property. A new method recently suggested, called by {\it signal lasso} (or its updating version: adaptive signal lasso) was proposed to solve the network reconstruction problem, where the signal parameter can be shrunk to either 0 or 1 in two different directions. The signal lasso or adaptive signal lasso employed the additive penalty of signal and non-signal terms which is a convex function and easily to complementation in computation. However their methods need tuning the one or two parameters to find an optimal solution, which is time cost for large size network. In this paper we propose new signal lasso method based on two penalty functions to estimate the signal parameter and uncovering network topology in complex network with a small amount of observations. The penalty functions we introduced are non-convex function, thus coordinate descent algorithms are suggested. We find in this method the tuning parameter can be set to a large enough values such that the signal parameter can be completely shrunk either 0 or 1. The extensive simulations are conducted in linear regression models with different assumptions, the evolutionary-game-based dynamic model and Kuramoto model of synchronization problem. The advantage and disadvantage of each method are fully discussed in various conditions. Finally a real example comes from behavioral experiment is used for illustration. Our results show that signal lasso with non-convex penalties is effective and fast in estimating signal parameters in linear regression model.

math.ST↗

arXiv2025

Vectorial microlasers with designable topological charges based on Möbius-like correspondence in quasi-BICs

The ability to control topological properties of laser emission represents a fundamental advancement in photonic technology. Achieving topological laser in a single compact photonic structure is crucial for device integration and miniaturization but faces significant challenges for designing both the high-quality (high-Q) mode and radiative topological configurations. Recently, bound states in the continuum (BICs), as extraordinary states possessing both ultrahigh Q factors and polarization topological charges, have been demonstrated as a promising platform for compact topological lasers. However, as the cornerstone of BIC lasing's non-trivial properties, topological charges of BICs are protected by real-space structural symmetries, which simultaneously impose fundamental limitations that hinder their designability of lasing topological charges. Here, we propose and experimentally demonstrate a compound cavity design method based on the Möbius-like correspondence in quasi-BICs (q-BICs), by which compact vectorial microlasers with designable topological charges can be realized. We reveal the hidden connection between real-space symmetry breaking and eigen-polarizations of q-BICs from the triangular photonic crystal (PhC) slab, manifesting as a Möbius-like correspondence. By splicing PhC slab sectors utilizing this Möbius-like correspondence, we establish a one-to-one correspondence between compound cavities and their lasing topological charges. Vectorial lasing with designable topological charges from $-5$ to $+5$ were experimentally realized. Our work establishes a novel BIC-based platform that enables designable topological lasing, providing a promising route toward compact topological sources.

physics.optics↗

arXiv2025

Conditional cross-fitting for unbiased machine-learning-assisted covariate adjustment in randomized experiments

Randomized experiments are the gold standard for estimating the average treatment effect (ATE). While covariate adjustment can reduce the asymptotic variances of the unbiased Horvitz-Thompson estimators for the ATE, it suffers from finite-sample biases due to data reuse in both prediction and estimation. Traditional sample-splitting and cross-fitting methods can address the problem of data reuse and obtain unbiased estimators. However, they require that the data are independently and identically distributed, which is usually violated under the design-based inference framework for randomized experiments. To address this challenge, we propose a novel conditional cross-fitting method, under the design-based inference framework, where potential outcomes and covariates are fixed and the randomization is the sole source of randomness. We propose sample-splitting algorithms for various randomized experiments, including Bernoulli randomized experiments, completely randomized experiments, and stratified randomized experiments. Based on the proposed algorithms, we construct unbiased covariate-adjusted ATE estimators and propose valid inference procedures. Our methods can accommodate flexible machine-learning-assisted covariate adjustments and allow for model misspecification.

stat.ME↗

arXiv2025

An Efficient Deep Learning Approach for Approximating Parameter-to-Solution Maps of PDEs

In this paper, we consider approximating the parameter-to-solution maps of parametric partial differential equations (PPDEs) using deep neural networks (DNNs). We propose an efficient approach combining reduced collocation methods (RCMs) and DNNs. In the approximation analysis section, we rigorously derive sharp upper bounds on the complexity of the neural networks. These bounds only depend on the reduced basis dimension rather than the high-fidelity discretization dimension, thereby theoretically guaranteeing the computational efficiency of our approach. In numerical experiments, we implement the RCM using radial basis function finite differences (RBF-FD) and proper orthogonal decomposition (POD), and propose the POD-DNN algorithm. We consider various types of PPDEs and compare the accuracy and efficiency of different solvers. The POD-DNN has demonstrated significantly accelerated inference speeds compared with conventional numerical methods owing to the offline-online computation strategy. Furthermore, by employing the reduced basis methods (RBMs), it also outperforms standard DNNs in computational efficiency while maintaining comparable accuracy.

math.NA↗

arXiv2025

Asymmetrical Filtering Impairments Mitigation for Digital- Subcarrier-Multiplexing Transmissions Enabled by Multiplication-free K-State Reserved Complex MLSE

We propose a multiplication-free K-state reserved complex maximum-likelihood-sequence-estimation (MLSE) to mitigate asymmetrical filtering impairments in digital-subcarrier-multiplexing transmissions. A required optical-to-noise ratio of 1.63dB over the conventional real MLSE is obtained after transmitting 90 GBaud DSCM DP-16QAM signal over 14 WSSs without multiplications.

physics.optics↗

arXiv2025

Spectral Algorithms under Covariate Shift

Spectral algorithms leverage spectral regularization techniques to analyze and process data, providing a flexible framework for addressing supervised learning problems. To deepen our understanding of their performance in real-world scenarios where the distributions of training and test data may differ, we conduct a rigorous investigation into the convergence behavior of spectral algorithms under covariate shift. In this setting, the marginal distributions of the input data differ between the training and test datasets, while the conditional distribution of the output given the input remains unchanged. Within a non-parametric regression framework over a reproducing kernel Hilbert space, we analyze the convergence rates of spectral algorithms under covariate shift and show that they achieve minimax optimality when the density ratios between the training and test distributions are uniformly bounded. However, when these density ratios are unbounded, the spectral algorithms may become suboptimal. To address this issue, we propose a novel weighted spectral algorithm with normalized weights that incorporates density ratio information into the learning process. Our theoretical analysis shows that this normalized weighted approach achieves optimal capacity-independent convergence rates, but the rates will suffer from the saturation phenomenon. Furthermore, by introducing a weight clipping technique, we demonstrate that the convergence rates of the weighted spectral algorithm with clipped weights can approach the optimal capacity-dependent convergence rates arbitrarily closely. This improvement resolves the suboptimality issue in unbounded density ratio scenarios and advances the state-of-the-art by refining existing theoretical results.

stat.ML↗

arXiv2025

Truncated Kernel Stochastic Gradient Descent on Spheres

Inspired by the structure of spherical harmonics, we propose the truncated kernel stochastic gradient descent (T-kernel SGD) algorithm with a least-square loss function for spherical data fitting. T-kernel SGD introduces a novel regularization strategy by implementing stochastic gradient descent through a closed-form solution of the projection of the stochastic gradient in a low-dimensional subspace. In contrast to traditional kernel SGD, the regularization strategy implemented by T-kernel SGD is more effective in balancing bias and variance by dynamically adjusting the hypothesis space during iterations. The most significant advantage of the proposed algorithm is that it can achieve theoretically optimal convergence rates using a constant step size (independent of the sample size) while overcoming the inherent saturation problem of kernel SGD. Additionally, we leverage the structure of spherical polynomials to derive an equivalent T-kernel SGD, significantly reducing storage and computational costs compared to kernel SGD. Typically, T-kernel SGD requires only $\mathcal{O}(n^{1+\frac{d}{d-1}ε})$ computational complexity and $\mathcal{O}(n^{\frac{d}{d-1}ε})$ storage to achieve optimal rates for the d-dimensional sphere, where $0<ε<\frac{1}{2}$ can be arbitrarily small if the optimal fitting or the underlying space possesses sufficient regularity. This regularity is determined by the smoothness parameter of the objective function and the decaying rate of the eigenvalues of the integral operator associated with the kernel function, both of which reflect the difficulty of the estimation problem. Our main results quantitatively characterize how this prior information influences the convergence of T-kernel SGD. The numerical experiments further validate the theoretical findings presented in this paper.

cs.LG↗

arXiv2025

Meron Spin Textures in Momentum Space Spawning from Bound States in the Continuum

Topological spin textures, such as merons and skyrmions, have shown significance in both fundamental science and practical applications across diverse physical systems. The optical skyrmionic textures in real space have been extensively explored, but which in momentum space are still rarely studied. Here, we report the experimental generation of momentum-space meron spin textures via bound states in the continuum (BICs) in photonic crystal slabs. We show that under circularly polarized illumination, the momentum-space vortex topology of BICs can transform light with meron spin textures in momentum space. These merons exhibit polarity-switchable configurations controlled by incident light polarization. We theoretically and experimentally verify the generation of momentum-space merons and demonstrate their operational flexibility across a broad spectral range. Our results establish a connection between different momentum-space topologies and provide a robust and compact platform for generating topological spin textures.

physics.optics↗

arXiv2025

Quantum oscillation in Hopf-link semimetals

Since the discovery of the relation between the Chern number and quantum Hall effect, searching for observables of topological invariants has been an intriguing topic. Topological Hopf-link semimetals have attracted tremendous interest, in which the conduction and valence energy bands touch at linked nodal lines. However, it is challenging to identify this sophisticated topology. We propose to use the quantum oscillation in strong magnetic fields to probe the Hopf links. For a generic model of Hopf-link semimetal that captures the linked-trivial phase transition, we figure out the phase shifts of oscillation for all Fermi pockets in all magnetic-field directions, by presenting self-consistent results from the Fermi surface tomography, Landau fan diagram, and electrical resistivity. As the magnetic field is rotated, the phase shifts exhibit a unique pattern, which could help to identify Hopf links in real materials, such as those in Li$_2$NaN.

cond-mat.mes-hall↗

arXiv2025

A Synergy Scoring Filter for Unsupervised Anomaly Detection with Noisy Data

Noise-inclusive fully unsupervised anomaly detection (FUAD) holds significant practical relevance. Although various methods exist to address this problem, they are limited in both performance and scalability. Our work seeks to overcome these obstacles, enabling broader adaptability of unsupervised anomaly detection (UAD) models to FUAD. To achieve this, we introduce the Synergy Scoring Filter (SSFilter), the first fully unsupervised anomaly detection approach to leverage sample-level filtering. SSFilter facilitates end-to-end robust training and applies filtering to the complete training set post-training, offering a model-agnostic solution for FUAD. Specifically, SSFilter integrates a batch-level anomaly scoring mechanism based on mutual patch comparison and utilizes regression errors in anomalous regions, alongside prediction uncertainty, to estimate sample-level uncertainty scores that calibrate the anomaly scoring mechanism. This design produces a synergistic, robust filtering approach. Furthermore, we propose a realistic anomaly synthesis method and an integrity enhancement strategy to improve model training and mitigate missed noisy samples. Our method establishes state-of-the-art performance on the FUAD benchmark of the recent large-scale industrial anomaly detection dataset, Real-IAD. Additionally, dataset-level filtering enhances the performance of various UAD methods on the FUAD benchmark, and the high scalability of our approach significantly boosts its practical applicability.

eess.IV↗

arXiv2025

MiniMaxAD: A Lightweight Autoencoder for Feature-Rich Anomaly Detection

Previous industrial anomaly detection methods often struggle to handle the extensive diversity in training sets, particularly when they contain stylistically diverse and feature-rich samples, which we categorize as feature-rich anomaly detection datasets (FRADs). This challenge is evident in applications such as multi-view and multi-class scenarios. To address this challenge, we developed MiniMaxAD, a efficient autoencoder designed to efficiently compress and memorize extensive information from normal images. Our model employs a technique that enhances feature diversity, thereby increasing the effective capacity of the network. It also utilizes large kernel convolution to extract highly abstract patterns, which contribute to efficient and compact feature embedding. Moreover, we introduce an Adaptive Contraction Hard Mining Loss (ADCLoss), specifically tailored to FRADs. In our methodology, any dataset can be unified under the framework of feature-rich anomaly detection, in a way that the benefits far outweigh the drawbacks. Our approach has achieved state-of-the-art performance in multiple challenging benchmarks. Code is available at: \href{https://github.com/WangFengJiee/MiniMaxAD}{https://github.com/WangFengJiee/MiniMaxAD}

cs.CV↗

arXiv2025

Berry-Esseen bounds for design-based causal inference with possibly diverging treatment levels and varying group sizes

Neyman (1923/1990) introduced the randomization model, which contains the notation of potential outcomes to define causal effects and a framework for large-sample inference based on the design of the experiment. However, the existing theory for this framework is far from complete, especially when the number of treatment levels diverges and the treatment group sizes vary. We provide a unified discussion of statistical inference under the randomization model with general treatment group sizes. We formulate the estimator in terms of a linear permutation statistic and use results based on Stein's method to derive various Berry--Esseen bounds on the linear and quadratic functions of the estimator. These new Berry--Esseen bounds serve as the basis for design-based causal inference with possibly diverging treatment levels and a diverging number of causal parameters of interest. We also fill an important gap by proposing novel variance estimators for experiments with possibly many treatment levels without replications. Equipped with the newly developed results, design-based causal inference in general settings becomes more convenient with stronger theoretical guarantees.

stat.ME↗

arXiv2025

SmartEraser: Remove Anything from Images using Masked-Region Guidance

Object removal has so far been dominated by the mask-and-inpaint paradigm, where the masked region is excluded from the input, leaving models relying on unmasked areas to inpaint the missing region. However, this approach lacks contextual information for the masked area, often resulting in unstable performance. In this work, we introduce SmartEraser, built with a new removing paradigm called Masked-Region Guidance. This paradigm retains the masked region in the input, using it as guidance for the removal process. It offers several distinct advantages: (a) it guides the model to accurately identify the object to be removed, preventing its regeneration in the output; (b) since the user mask often extends beyond the object itself, it aids in preserving the surrounding context in the final result. Leveraging this new paradigm, we present Syn4Removal, a large-scale object removal dataset, where instance segmentation data is used to copy and paste objects onto images as removal targets, with the original images serving as ground truths. Experimental results demonstrate that SmartEraser significantly outperforms existing methods, achieving superior performance in object removal, especially in complex scenes with intricate compositions.

cs.CV↗