Arrow Research search

Author name cluster

Sheng Gao

Possible papers associated with this exact author name in Arrow. This page groups case-insensitive exact name matches and is not a full identity disambiguation profile.

8 papers
2 author rows

Possible papers

8

AAAI Conference 2026 Conference Paper

Safe Multi-Agent Reinforcement Learning via Distributional Safety Critic and Maximum Entropy Optimization

  • Qiwei Liu
  • Ye Yuan
  • Lingyue Zhang
  • Kaitian Chen
  • Yunkai Lv
  • Sheng Gao
  • Huaicheng Yan

Deploying multi-agent reinforcement learning (MARL) in safety-critical systems faces significant challenges due to insufficient agent exploration and inadequate safety constraint guarantees. Current approaches are constrained by two fundamental limitations: inefficient exploration leading to suboptimal policies, and expected-cost-based constraint frameworks failing to ensure full-process safety. To address these challenges, this paper proposes a novel safety-aware maximum entropy MARL framework using Conditional Value-at-Risk (CVaR) as a joint safety metric, which quantifies constraint satisfaction under worst-case scenarios for multi-agent systems. Moreover, we develop the Worst-Case Multi-Agent Soft Actor-Critic (WCMASAC) algorithm, incorporating sequential update mechanisms and maximum entropy optimization for heterogeneous agents, enhanced with distributed safety critics. Theoretically, we establish the monotonic improvement property, guaranteed constraint satisfaction, and convergence to a generalized Nash equilibrium for WCMASAC. Extensive experiments on Safety-Gymnasium based benchmarks demonstrate that WCMASAC outperforms state-of-the-art baselines in both task reward acquisition and safety constraint violation reduction, while exhibiting superior exploration efficiency and risk-aware control capabilities.

EAAI Journal 2025 Journal Article

Toward accurate and efficient weather prediction using a dual-gated spatiotemporal attention network

  • Zongwei Zhang
  • Lianlei Lin
  • Sheng Gao
  • Junkai Wang
  • Hanqing Zhao

Accurate weather forecasting plays a vital role in safeguarding human activities, mitigating the risks of extreme climate events, and supporting environmental policy and disaster preparedness. However, existing data-driven approaches often struggle to effectively model the complex spatiotemporal dynamics and multivariate dependencies inherent in meteorological systems, limiting their reliability and scalability. To address these challenges, we propose a novel dual-gated spatiotemporal attention network (DSANet) for multivariate weather prediction. DSANet integrates a convolutional self-attention hybrid module to jointly capture local and global spatial features, and a dual-gated channel-time module to model temporal patterns and inter-variable relationships. A wavelet-guided composite loss function is introduced to enhance prediction accuracy in fluctuating and dynamic weather regions. Extensive experiments on both global and regional datasets demonstrate that DSANet outperforms baseline models in terms of accuracy, with a mean absolute error of 1. 78 K in 3-day lead-time global temperature forecasting. In addition, DSANet exhibits strong generalization and fast inference, making it well-suited for real-time and off-site forecasting. By significantly improving the accuracy, efficiency, and transferability of multivariate weather forecasting, DSANet provides a scalable and effective tool for next-generation climate intelligence and decision-making support systems.

EAAI Journal 2023 Journal Article

A real-time mechanical fault diagnosis approach based on lightweight architecture search considering industrial edge deployments

  • Sihan Ma
  • Hongchun Sun
  • Sheng Gao
  • Guixing Zhou

Mechanical intelligence diagnostic models based on deep learning have increased diagnostic accuracy. However, the industrial application of deep-learning models is hindered by the extra memory requirement and the increasing computational overhead of deep-learning models, which limit edge deployment and reduce real-time performance, respectively. Therefore, we propose a mechanical fault diagnosis strategy based on a multi-objective automatic optimization architecture search that combines variable layer search, efficient search space, and a real-time search strategy. A lightweight diagnostic network that integrates the computing requirements of edge deployment, real-time testing, and accuracy is constructed. The proposed model considers the influence of compound faults and the strong noise of rotating machinery in engineering applications. The experiments on the defect data of two sets of rotating equipment show that our model is more lightweight and has higher accuracy than other advanced baselines.

NeurIPS Conference 2023 Conference Paper

DP-HyPO: An Adaptive Private Framework for Hyperparameter Optimization

  • Hua Wang
  • Sheng Gao
  • Huanyu Zhang
  • Weijie Su
  • Milan Shen

Hyperparameter optimization, also known as hyperparameter tuning, is a widely recognized technique for improving model performance. Regrettably, when training private ML models, many practitioners often overlook the privacy risks associated with hyperparameter optimization, which could potentially expose sensitive information about the underlying dataset. Currently, the sole existing approach to allow privacy-preserving hyperparameter optimization is to uniformly and randomly select hyperparameters for a number of runs, subsequently reporting the best-performing hyperparameter. In contrast, in non-private settings, practitioners commonly utilize "adaptive" hyperparameter optimization methods such as Gaussian Process-based optimization, which select the next candidate based on information gathered from previous outputs. This substantial contrast between private and non-private hyperparameter optimization underscores a critical concern. In our paper, we introduce DP-HyPO, a pioneering framework for "adaptive" private hyperparameter optimization, aiming to bridge the gap between private and non-private hyperparameter optimization. To accomplish this, we provide a comprehensive differential privacy analysis of our framework. Furthermore, we empirically demonstrate the effectiveness of DP-HyPO on a diverse set of real-world datasets.

JMLR Journal 2023 Journal Article

Sparse GCA and Thresholded Gradient Descent

  • Sheng Gao
  • Zongming Ma

Generalized correlation analysis (GCA) is concerned with uncovering linear relationships across multiple data sets. It generalizes canonical correlation analysis that is designed for two data sets. We study sparse GCA when there are potentially multiple leading generalized correlation tuples in data that are of interest and the loading matrix has a small number of nonzero rows. It includes sparse CCA and sparse PCA of correlation matrices as special cases. We first formulate sparse GCA as a generalized eigenvalue problem at both population and sample levels via a careful choice of normalization constraints. Based on a Lagrangian form of the sample optimization problem, we propose a thresholded gradient descent algorithm for estimating GCA loading vectors and matrices in high dimensions. We derive tight estimation error bounds for estimators generated by the algorithm with proper initialization. We also demonstrate the prowess of the algorithm on a number of synthetic data sets. [abs] [ pdf ][ bib ] &copy JMLR 2023. ( edit, beta )

AAAI Conference 2015 Conference Paper

VecLP: A Realtime Video Recommendation System for Live TV Programs

  • Sheng Gao
  • Dai Zhang
  • Honggang Zhang
  • Chao Huang
  • Yongsheng Zhang
  • Jianxin Liao
  • Jun Guo

We propose VecLP, a novel Internet Video recommendation system working for Live TV Programs in this paper. Given little information on the live TV programs, our proposed VecLP system can effectively collect necessary information on both the programs and the subscribers as well as a large volume of related online videos, and then recommend the relevant Internet videos to the subscribers. For that, the key frames are firstly detected from the live TV programs, and then visual and textual features are extracted from these frames to enhance the understanding of the TV broadcasts. Furthermore, by utilizing the subscribers’ profiles and their social relationships, a user preference model is constructed, which greatly improves the diversity of the recommendations in our system. The subscriber’s browsing history is also recorded and used to make a further personalized recommendation. This work also illustrates how our proposed VecLP system makes it happen. Finally, we dispose some sort of new recommendation strategies in use at the system to meet special needs from diverse live TV programs and throw light upon how to fuse these strategies.

v2026.09.13