← Search

Michal Lukasik

11 accepted papers

2026

REAL: Regression-Aware Reinforcement Learning for LLM-as-a-Judge

ICML 2026poster

Large language models (LLMs) are increasingly deployed as automated evaluators that assign numeric scores to model outputs, a paradigm known as LLM-as-a-Judge. However, standard Reinforcement Learning (RL) methods typically rely on binary rewards (e.g., 0-1 accuracy), thereby ignoring the ordinal st…

Cited by 0SourceScholar
2025

Better autoregressive regression with LLMs via regression-aware fine-tuning

ICLR 2025spotlight

Decoder-based large language models (LLMs) have proven highly versatile, with remarkable successes even on problems ostensibly removed from traditional language generation. One such example is solving regression problems, where the targets are real numbers rather than textual tokens. A common appr…

Cited by 0SourcePDFScholar
2025

Bipartite Ranking From Multiple Labels: On Loss Versus Label Aggregation

ICML 2025poster

Bipartite ranking is a fundamental supervised learning problem, with the goal of learning a ranking over instances with maximal area under the ROC curve (AUC) against a single binary target label. However, one may often observe multiple binary target labels, e.g., from distinct human annotators. How…

Cited by 1SourcePDFScholar
2025

TRACT: Regression-Aware Fine-tuning Meets Chain-of-Thought Reasoning for LLM-as-a-Judge

ACL 2025long

The LLM-as-a-judge paradigm uses large language models (LLMs) for automated text evaluation, assigning a score to the input based on scoring rubrics. Existing methods for fine-tuning LLM-as-a-judge use cross-entropy (CE) loss, which neglects the numeric nature of score prediction. Recent work addres…

2024

On Bias-Variance Alignment in Deep Models

ICLR 2024spotlight

Classical wisdom in machine learning holds that the generalization error can be decomposed into bias and variance, and these two terms exhibit a \emph{trade-off}. However, in this paper, we show that for an ensemble of deep learning based classification models, bias and variance are \emph{aligned} a…

Cited by 3SourcePDFScholar
2024

Regression Aware Inference with LLMs

EMNLP 2024finding

Large language models (LLMs) have shown strong results on a range of applications, including regression and scoring tasks.Typically, one obtains outputs from an LLM via autoregressive sampling from the model’s output distribution. We show that this inference strategy can be sub-optimal for common re…

Cited by 3SourcePDFScholar
2024

Two-stage LLM Fine-tuning with Less Specialization and More Generalization

ICLR 2024poster

Pretrained large language models (LLMs) are general purpose problem solvers applicable to a diverse set of tasks with prompts. They can be further improved towards a specific task by fine-tuning on a specialized dataset. However, fine-tuning usually makes the model narrowly specialized on this datas…

Cited by 6SourcePDFScholar
2023

Large Language Models with Controllable Working Memory

ACL 2023findings

Large language models (LLMs) have led to a series of breakthroughs in natural language processing (NLP), partly owing to the massive amounts of world knowledge they memorize during pretraining. While many downstream applications provide the model with an informational context to aid its underlying t…

Cited by 148SourcePDFScholar
2023

ResMem: Learn what you can and memorize the rest

NeurIPS 2023poster

The impressive generalization performance of modern neural networks is attributed in part to their ability to implicitly memorize complex training patterns. Inspired by this, we explore a novel mechanism to improve model generalization via explicit memorization. Specifically, we propose the residual…

Cited by 12SourcePDFScholar
2023

Robust distillation for worst-class performance: on the interplay between teacher and student objectives

UAI 2023poster

Knowledge distillation is a popular technique that has been shown to produce remarkable gains in average accuracy. However, recent work has shown that these gains are not uniform across subgroups in the data, and can often come at the cost of accuracy on rare subgroups and classes. Robust optimizati…

Cited by 10SourcePDFScholar