Juho Lee

26

Papers

82

Total Citations

Papers (26)

Learning Diverse Attacks on Large Language Models for Robust Red-Teaming and Safety Tuning

HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Models

Finite-Dimensional BFRY Priors and Variational Bayesian Inference for Power Law Models

Lipsum-FT: Robust Fine-Tuning of Zero-Shot Models Using Random Text Guidance

Dimension Agnostic Neural Processes

Bayesian Neural Scaling Law Extrapolation with Prior-Data Fitted Networks

Variational Partial Group Convolutions for Input-Aware Partial Equivariance of Rotations and Color-Shifts

Learning to Explore for Stochastic Gradient MCMC

A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models

SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data

A Multi-Mode Modulator for Multi-Domain Few-Shot Classification

Tree-Guided MCMC Inference for Normalized Random Measure Mixture Models

Spear and Shield: Adversarial Attacks and Defense Methods for Model-Based Link Prediction on Continuous-Time Dynamic Graphs

Variational Bayesian Pseudo-Coreset

Uncertainty-Aware Attention for Reliable Interpretation and Prediction

DropMax: Adaptive Variational Softmax

Bootstrapping neural processes

Neural Complexity Measures

Diversity Matters When Learning From Ensembles

Mini-Batch Consistent Slot Set Encoder for Scalable Set Encoding

On Divergence Measures for Bayesian Pseudocoresets

Set-based Meta-Interpolation for Few-Task Meta-Learning

Function Space Bayesian Pseudocoreset for Bayesian Neural Networks

Bayesian inference on random simple graphs with power law degree distributions

Beyond the Chinese Restaurant and Pitman-Yor processes: Statistical Models with double power-law behavior

Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks