← Search

Shaolei Ren

16 accepted papers

2024

ArchLock: Locking DNN Transferability at the Architecture Level with a Zero-Cost Binary Predictor

ICLR 2024poster

Deep neural network (DNN) models, despite their impressive performance, are vulnerable to exploitation by attackers who attempt to transfer them to other tasks for their own benefit. Current defense strategies mainly address this vulnerability at the model parameter level, leaving the potential of a…

2024

Bileve: Securing Text Provenance in Large Language Models Against Spoofing with Bi-level Signature

NeurIPS 2024poster

Text watermarks for large language models (LLMs) have been commonly used to identify the origins of machine-generated content, which is promising for assessing liability when combating deepfake or harmful content. While existing watermarking techniques typically prioritize robustness against removal…

2023

Anytime-Competitive Reinforcement Learning with Policy Prior

NeurIPS 2023poster

This paper studies the problem of Anytime-Competitive Markov Decision Process (A-CMDP). Existing works on Constrained Markov Decision Processes (CMDPs) aim to optimize the expected reward while constraining the expected cost over random dynamics, but the cost in a specific episode can still be unsat…

Cited by 2SourcePDFScholar
2023

Beyond Black-Box Advice: Learning-Augmented Algorithms for MDPs with Q-Value Predictions

NeurIPS 2023poster

We study the tradeoff between consistency and robustness in the context of a single-trajectory time-varying Markov Decision Process (MDP) with untrusted machine-learned advice. Our work departs from the typical approach of treating advice as coming from black-box sources by instead considering a set…

Cited by 4SourcePDFScholar
2023

Learning for Edge-Weighted Online Bipartite Matching with Robustness Guarantees

ICML 2023poster

Many problems, such as online ad display, can be formulated as online bipartite matching. The crucial challenge lies in the nature of sequentially-revealed online item information, based on which we make irreversible matching decisions at each step. While numerous expert online algorithms have been…

2023

Learning-Assisted Algorithm Unrolling for Online Optimization with Budget Constraints

AAAI 2023technical

Online optimization with multiple budget constraints is challenging since the online decisions over a short time horizon are coupled together by strict inventory constraints. The existing manually-designed algorithms cannot achieve satisfactory average performance for this setting because they often…

Cited by 3SourcePDFScholar
2023

NNSplitter: An Active Defense Solution for DNN Model via Automated Weight Obfuscation

ICML 2023poster

As a type of valuable intellectual property (IP), deep neural network (DNN) models have been protected by techniques like watermarking. However, such passive model protection cannot fully prevent model abuse. In this work, we propose an active model IP protection scheme, namely NNSplitter, which act…

2023

Robust Learning for Smoothed Online Convex Optimization with Feedback Delay

NeurIPS 2023poster

We study a general form of Smoothed Online Convex Optimization, a.k.a. SOCO, including multi-step switching costs and feedback delay. We propose a novel machine learning (ML) augmented online algorithm, Robustness-Constrained Learning (RCL), which combines untrusted ML predictions with a trusted exp…

Cited by 4SourcePDFScholar
2022

Informed Learning by Wide Neural Networks: Convergence, Generalization and Sampling Complexity

ICML 2022spotlight

By integrating domain knowledge with labeled samples, informed machine learning has been emerging to improve the learning performance for a wide range of applications. Nonetheless, rigorous understanding of the role of injected domain knowledge has been under-explored. In this paper, we consider an…

Cited by 4SourcePDFScholar
2022

Navigating Memory Construction by Global Pseudo-Task Simulation for Continual Learning

NeurIPS 2022accept

Continual learning faces a crucial challenge of catastrophic forgetting. To address this challenge, experience replay (ER) that maintains a tiny subset of samples from previous tasks has been commonly used. Existing ER works usually focus on refining the learning objective for each task with a stati…