AI Sweden logo

Master Thesis: Cross-Lingual Safety and System Identity Alignment via DPO

AI Sweden

Stockholm, SE; Gothenburg, SEHybridPosted 31 days ago
InternshipInternResearch & ScientistResearch Institute
Original posting ↗

Summary

Open-weight multilingual models should maintain a consistent system identity and appropriate safety behaviour across languages. However, safety alignment can vary across languages and may introduce over-refusal, particularly in lower-resource settings.

Job description

Introduction

Open-weight multilingual models should maintain a consistent system identity and appropriate safety behaviour across languages. However, safety alignment can vary across languages and may introduce over-refusal, particularly in lower-resource settings.

Project Background and Problem Statement

We have developed the openeurollm-model-identity dataset and a multilingual safety refusal suite. This thesis investigates DPO-based alignment in Swedish and Icelandic, focusing on three questions:

  • System identity: Does the model reliably identify itself as OpenEuroLLM and resist attempts to overwrite that identity?
  • Safety alignment: Does the model appropriately refuse genuinely unsafe requests across languages?
  • Over-refusal: Does safety alignment cause the model to refuse legitimate requests?

The broader question is whether these behaviours can be improved without degrading general instruction-following performance.

Outline

  • Literature study: Review cross-lingual safety alignment, over-refusal, model identity, and preference optimization.
  • Data analysis: Assess the quality and suitability of the preference data, including both preference correctness and the absolute quality of chosen responses.
  • Implementation: Fine-tune Prelude 9B using identity and safety preference data with Hugging Face TRL or Megatron-LM. Compare identity-only, safety-only, and combined training where feasible.
  • Evaluation: Compare the resulting models against the baseline on identity consistency, safety and jailbreak robustness, over-refusal on benign queries, and general instruction-following performance in Swedish and Icelandic.

Who we're looking for

We are seeking curious, self-driven MSc students eager to work at the frontier of open-weight European AI research (LLMs). You thrive on empirical discovery, design rigorous experiments, and let data challenge your assumptions.

  • Ongoing Master's studies in Computer Science, Data Science, Machine Learning, Engineering Physics, or a related quantitative field.
  • Proficiency in Python and hands-on experience with modern deep learning frameworks (PyTorch, Hugging Face ecosystem).
  • Familiarity with LLM post-training alignment (e.g., SFT, DPO, RLHF/RLVR) or context-extension, alongside comfort running distributed GPU training in Linux/HPC environments.

Requirements

  • Ongoing Master's studies in Computer Science, Data Science, Machine Learning, Engineering Physics, or a related quantitative field
  • Proficiency in Python
  • Hands-on experience with modern deep learning frameworks (PyTorch, Hugging Face ecosystem)
  • Familiarity with LLM post-training alignment (e.g., SFT, DPO, RLHF/RLVR) or context-extension
  • Comfort running distributed GPU training in Linux/HPC environments

Company overview

AI Sweden is Sweden's national center for applied artificial intelligence, established in 2019 with a mission to accelerate the adoption and implementation of AI technologies for societal benefit, economic competitiveness, and national advancement. As a collaborative research institute, it brings together over 170 partners from diverse sectors including government agencies, universities, research institutions, startups, and global corporations. The organization operates as a catalyst for AI innovation, focusing on practical applications across healthcare, energy systems, municipalities, and various industries. With a core team of over 80 professionals comprising AI experts, data scientists, research engineers, linguists, mathematicians, policy experts, and change leaders, AI Sweden facilitates knowledge sharing, talent development, and infrastructure building. The institute operates on a not-for-profit basis, funded by Sweden's innovation agency Vinnova, the Swedish Agency for Economic and Regional Growth, European Regional Development Fund, Västra Götalandsregionen, and partner contributions. Through its AI Labs, talent programs, workshops, and collaborative projects, AI Sweden addresses critical societal challenges while bridging the gap between cutting-edge research and real-world implementation, positioning Sweden as a leader in responsible and impactful AI development.

Explore more jobs of your matchFind more biopharma roles with GetBioJob.