← Search

Waleed Mustafa

8 accepted papers

2026

TORA: Train Once, Realign Anytime for Offline Multi-Objective Reinforcement Learning

AAAI 2026technical

Intelligent agents in real-world applications must adapt their behavior to changing contexts and user preferences. For example, planning a road trip requires considering both travel time and cost. Multi-objective reinforcement learning (MORL) provides a principled approach to navigate such trade-of

Cited by 0SourcePDFScholar
2025

Mitigating Spurious Features in Contrastive Learning with Spectral Regularization

NeurIPS 2025poster

Neural networks generally prefer simple and easy-to-learn features. When these features are spuriously correlated with the labels, the network's performance can suffer, particularly for underrepresented classes or concepts. Self-supervised representation learning methods, such as contrastive learnin…

Cited by 0SourcecodeScholar
2024

Ethics in Action: Training Reinforcement Learning Agents for Moral Decision-making In Text-based Adventure Games

AISTATS 2024poster

Reinforcement Learning (RL) has demonstrated its potential in solving goal-oriented sequential tasks. However, with the increasing capabilities of RL agents, ensuring morally responsible agent behavior is becoming a pressing concern. Previous approaches have included moral considerations by statical…

2024

Interpretable Tensor Fusion

IJCAI 2024poster

Conventional machine learning methods are predominantly designed to predict outcomes based on a single data type. However, practical applications may encompass data of diverse types, such as text, images, and audio. We introduce interpretable tensor fusion (InTense), a multimodal learning method tra…

Cited by 2SourcePDFScholar
2024

Non-vacuous Generalization Bounds for Adversarial Risk in Stochastic Neural Networks

AISTATS 2024poster

Adversarial examples are manipulated samples used to deceive machine learning models, posing a serious threat in safety-critical applications. Existing safety certificates for machine learning models are limited to individual input examples, failing to capture generalization to unseen data. To addre…

2021

Fine-grained Generalization Analysis of Structured Output Prediction

IJCAI 2021poster

In machine learning we often encounter structured output prediction problems (SOPPs), i.e. problems where the output space admits a rich internal structure. Application domains where SOPPs naturally occur include natural language processing, speech recognition, and computer vision. Typical SOPPs hav…

Cited by 6SourcePDFScholar
2021

Norm-Based Generalisation Bounds for Deep Multi-Class Convolutional Neural Networks

AAAI 2021technical

We show generalisation error bounds for deep learning with two main improvements over the state of the art. (1) Our bounds have no explicit dependence on the number of classes except for logarithmic factors. This holds even when formulating the bounds in terms of the Frobenius-norm of the weight mat…

Cited by 36SourcePDFScholar