← Search

Yue Song

25 accepted papers

2026

Constitutional Classifiers++: Production-Grade Defenses against Universal Jailbreaks

ICLR 2026poster

We introduce enhanced Constitutional Classifiers that deliver production-grade jailbreak robustness with dramatically reduced computational costs and refusal rates compared to previous-generation defenses. We first identify vulnerabilities in existing systems that evaluate model outputs without rega…

Cited by 0SourceScholar
2026

Fast and Stable Riemannian Metrics on SPD Manifolds via Cholesky Product Geometry

ICLR 2026poster

Recent advances in Symmetric Positive Definite (SPD) matrix learning show that Riemannian metrics are fundamental to effective SPD neural networks. Motivated by this, we revisit the geometry of the Cholesky factors and uncover a simple product structure that enables convenient metric design. Buildin…

Cited by 0SourcecodeScholar
2026

Hilbert Curve-Based Attention Enabling Topology-Preserving Image Tensor Representation for Semantic Segmentation Network

CVPR 2026

Drone-based building defect segmentation remains challenging due to complex surface textures and illumination variations. We propose TPSegformer, a topology-preserving segmentation framework that mitigates mis-segmentation in such scenarios. Its decoder incorporates a Hilbert curve-based topology-pr

Cited by 0SourcecodeScholar
2025

Kuramoto Orientation Diffusion Models

NeurIPS 2025poster

Orientation-rich images, such as fingerprints and textures, often exhibit coherent angular directional patterns that are challenging to model using standard generative approaches based on isotropic Euclidean diffusion. Motivated by the role of phase synchronization in biological systems, we propose…

Cited by 0SourceScholar
2025

Understanding Matrix Function Normalizations in Covariance Pooling through the Lens of Riemannian Geometry

ICLR 2025poster

Global Covariance Pooling (GCP) has been demonstrated to improve the performance of Deep Neural Networks (DNNs) by exploiting second-order statistics of high-level representations. GCP typically performs classification of the covariance matrices by applying matrix function normalization, such as mat…

2025

Unsupervised Region-Based Image Editing of Denoising Diffusion Models

AAAI 2025technical

Although diffusion models have achieved remarkable success in the field of image generation, their latent space remains under-explored. Current methods for identifying semantics within latent space often rely on external supervision, such as textual information and segmentation masks. In this paper,…

Cited by 0SourcePDFScholar
2024

Aligning Large Language Models with Representation Editing: A Control Perspective

NeurIPS 2024poster

Aligning large language models (LLMs) with human objectives is crucial for real-world applications. However, fine-tuning LLMs for alignment often suffers from unstable training and requires substantial computing resources. Test-time alignment techniques, such as prompting and guided decoding, do not…

2024

Navigating Chemical Space with Latent Flows

NeurIPS 2024poster

Recent progress of deep generative models in the vision and language domain has stimulated significant interest in more structured data generation such as molecules. However, beyond generating new random molecules, efficient exploration and a comprehensive understanding of the vast chemical space ar…

2024

RMLR: Extending Multinomial Logistic Regression into General Geometries

NeurIPS 2024poster

Riemannian neural networks, which extend deep learning techniques to Riemannian spaces, have gained significant attention in machine learning. To better classify the manifold-valued features, researchers have started extending Euclidean multinomial logistic regression (MLR) into Riemannian manifolds…

2024

Riemannian Multinomial Logistics Regression for SPD Neural Networks

CVPR 2024poster

Deep neural networks for learning Symmetric Positive Definite (SPD) matrices are gaining increasing attention in machine learning. Despite the significant progress most existing SPD networks use traditional Euclidean classifiers on an approximated space rather than intrinsic classifiers that accurat…

2023

Latent Traversals in Generative Models as Potential Flows

ICML 2023poster

Despite the significant recent progress in deep generative models, the underlying structure of their latent spaces is still poorly understood, thereby making the task of performing semantically meaningful latent traversals an open research challenge. Most prior work has aimed to solve this challenge…

2023

Masked Jigsaw Puzzle: A Versatile Position Embedding for Vision Transformers

CVPR 2023poster

Position Embeddings (PEs), an arguably indispensable component in Vision Transformers (ViTs), have been shown to improve the performance of ViTs on many vision tasks. However, PEs have a potentially high risk of privacy leakage since the spatial information of the input patches is exposed. This cave…

2022

GBA: A Tuning-free Approach to Switch between Synchronous and Asynchronous Training for Recommendation Models

NeurIPS 2022accept

High-concurrency asynchronous training upon parameter server (PS) architecture and high-performance synchronous training upon all-reduce (AR) architecture are the most commonly deployed distributed training modes for recommendation models. Although synchronous AR training is designed to have higher…

Cited by 3SourcePDFScholar
2021

Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance Pooling?

ICCV 2021poster

Global Covariance Pooling (GCP) aims at exploiting the second-order statistics of the convolutional feature. Its effectiveness has been demonstrated in boosting the classification performance of Convolutional Neural Networks (CNNs). Singular Value Decomposition (SVD) is used in GCP to compute the ma…

Cited by 28PDFcodeScholar
2020

Community-Centric Graph Convolutional Network for Unsupervised Community Detection

IJCAI 2020poster

Community detection, aiming at partitioning a network into multiple substructures, is practically importance. Graph convolutional network (GCN), a new deep-learning technique, has recently been developed for community detection. Markov Random Fields (MRF) has been combined with GCN in the MRFasGCN m…

Cited by 0SourcePDFScholar