Master Thesis: Intepretable Latent Dynamics World Models using KANs

Posted Yesterday
Be an Early Applicant
Stockholm, SWE
In-Office
Entry level
Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
We are shaping the future of digital connectivity the world relies on.
The Role
Conduct a 30-credit master's thesis developing interpretable latent-dynamics world models using KANs. Implement and train KAN-parameterized state-space models, validate them in partially observable Gymnasium environments, apply them to Ericsson RAN data, and benchmark against recurrent and interpretable models. Analyze learned dynamics symbolically and using control theory, then present results and deliver reproducible experiments.
Summary Generated by Built In
Join our Team
About this opportunity:
Modern mobile networks are increasingly operated by autonomous control loops. Before an automated agent is allowed to adjust a live network, its proposed actions must be evaluated, and world models are one way to achieve this.
For critical telecommunications infrastructure, predictive accuracy alone is not enough. Operators and regulators increasingly demand models whose behaviour can be inspected, audited, and justified. At Ericsson Research, we are exploring intrinsically interpretable machine learning for network dynamics. Kolmogorov-Arnold Networks (KANs) have demonstrated the potential to balance performance, interpretability, and model size.
Black-box sequence models, such as RNNs and transformers, often outperform feedforward predictors on network data because they exploit information carried in the trajectory history. This suggests that the observed indicators have strong temporal correlations.
The goal of this thesis is to investigate whether this advantage can be recovered inside an interpretable model: a nonlinear state-space model with a low-dimensional learned latent state, parameterised entirely by KANs. The full learned realisation-including the state transition and observation maps-should be possible to prune and express in closed form.
The thesis will be conducted by one student, corresponds to 30 hp, and is expected to start in early January 2027. The location is Stockholm, Sweden. Supervisors are Agustín Valencia and Maxime Bouton.
What you will do:
  • Formulate and implement a KAN-parameterised state-space model with a learned hidden state, trained on multi-step rollouts.
  • Validate the approach under controlled partial observability in standard Gymnasium environments, where hidden variables are known and can be masked. This will allow recovery of latent structure to be measured against ground truth.
  • Apply the validated method to a real Ericsson Radio Access Network dataset.
  • Benchmark the approach against recurrent models and state-of-the-art interpretable baselines.
  • Analyse the learned model symbolically through closed-form extraction and from a control-theoretic perspective.
  • Present the results to the Ericsson research team and deliver a repository with reproducible experiments.

The skills you bring:
  • You are a final-year Master's student in Mathematical Statistics, Applied Mathematics, Engineering Physics, Electrical Engineering, Computer Science, or a related programme.
  • You have a solid foundation in machine learning and probability or statistics.
  • You have strong Python skills and experience with a deep-learning framework, preferably PyTorch.
  • You have coursework or project experience in at least one of the following areas: dynamical systems, control theory, system identification, or time-series analysis.
  • You can work independently, structure an open research question, and communicate your findings clearly.

Why join Ericsson?At Ericsson, you'll have an outstanding opportunity. The chance to use your skills and imagination to push the boundaries of what's possible. To build solutions never seen before to some of the world's toughest problems. You'll be challenged, but you won't be alone. You'll be joining a team of diverse innovators, all driven to go beyond the status quo to craft what comes next.
What happens once you apply? Click Here to find all you need to know about what our typical hiring process looks like.Encouraging a diverse and inclusive organization is core to our values at Ericsson, that's why we champion it in everything we do. We truly believe that by collaborating with people with different experiences we drive innovation, which is essential for our future growth. We encourage people from all backgrounds to apply and realize their full potential as part of our Ericsson team. Ericsson is proud to be an Equal Opportunity Employer. learn more.
Primary country and city: Sweden (SE) || Stockholm
Req ID: 791197

Skills Required

  • Final-year Master's student in Mathematical Statistics, Applied Mathematics, Engineering Physics, Electrical Engineering, Computer Science, or a related program
  • Solid foundation in machine learning and probability or statistics
  • Strong Python skills
  • Experience with a deep-learning framework, preferably PyTorch
  • Coursework or project experience in dynamical systems, control theory, system identification, or time-series analysis
  • Ability to work independently, structure an open research question, and communicate findings clearly

What the Team is Saying

Vishal
Mayank
Granville
Sneha
Olga
Emily
Nicole
Sola
Aditi
Ericsson
Ericsson
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Stockholm
88,000 Employees
Year Founded: 1876

What We Do

Ericsson builds the digital connectivity the world relies on. Our technology underpins the mobile networks, platforms, and systems that billions of people, businesses, and societies depend on every day. We are a global leader in communications technology, delivering mobile network infrastructure, cloud software, and wireless connectivity solutions for service providers and enterprises worldwide. Our networks support connectivity across 180+ countries, helping power everyday communication as well as critical digital services at global scale. Connectivity has evolved far beyond consumer mobile use. Today, nearly 80% of the world’s population accesses the internet via mobile networks, and Ericsson is helping shape what comes next. We are advancing 5G and 5G Advanced, developing network APIs that open connectivity to the global developer ecosystem, and applying automation and AI to make networks more intelligent, efficient, and resilient. Ericsson was the first company to launch live 5G networks on five continents, and our 5G platform is now commercially live in 150+ networks across 60+ countries. We also support more than 36,000 enterprise customers, enabling secure, high-performance connectivity for industries such as manufacturing, aviation, logistics, utilities, and public safety, where reliability and performance are mission critical. Innovation is central to how we work. Ericsson has approximately 28,000 employees in research and development, backed by one of the strongest intellectual property portfolios in the industry with 60,000+ granted patents. Our engineers, researchers, and technologists work across 100+ global R&D sites, helping define how networks evolve and how digital infrastructure is built for the long term. As the world moves toward a mobile-first, AI-powered, and cloud-driven future, connectivity becomes the foundation for digital transformation across every industry. Ericsson is building that foundation, shaping the future of digital connectivity through technology that operates at global scale and supports real-world impact, today and for what comes next.

Why Work With Us

Ericsson is a place for people who want to work on technology that powers everyday life. You’ll contribute to large-scale systems used every day, tackle complex challenges in live environments, and keep developing your skills and career in your own vision.

Gallery

Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery

Ericsson Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Ericsson adopts a hybrid work model globally because we know balance matters. Sometimes things are better in real life. Other times we can be more productive at home. Our hybrid approach gives you the best of both worlds.

Typical time on-site: Flexible
Company Office Image
HQEricsson HQ - Kista, Sweden
Yokohama Office
Mexico City - Mexico Head Office
Indaiatuba-SP 5G Smart Factory
Athlone Software Campus
Austin ASIC Design Center
Beijing - China Headquarters
Bellevue Office
Bengaluru Hub
Boise - Ericsson Enterprise Wireless Solutions (formerly Cradlepoint) Headquarters
Bucharest Main Site
Budapest - Ericsson House
Company Office Image
Chennai Hub
Gurugram Hub - Ericsson India Global Services (Delhi NCR)
Company Office Image
Gurgaon Head Office - Ericsson India (Delhi NCR)
Irvine Office
Ottawa R&D Site
Kolkata Hub
Krakow R&D Center
Company Office Image
Lewisville 5G Smart Factory
Łódź R&D Site
Montreal AI & R&D Hub
Company Office Image
Morristown Office
Nanjing R&D & Manufacturing Hub
Noida Hub (Delhi NCR)
Overland Park Office
Plano Office - US Headquarters
Pune Hub
Company Office Image
Santa Clara D-15 Innovation Center
São Paulo - Brazil Main Office
Tokyo Head Office
Learn more

Similar Jobs

Ericsson Logo Ericsson

Verification Engineer

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Stockholm, SWE
88000 Employees

Ericsson Logo Ericsson

Master Thesis: AI-Assisted Root Cause Analysis for Cloud RAN

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Stockholm, SWE
88000 Employees

Ericsson Logo Ericsson

Master Thesis: Evaluation of Multipath Routing Algorithms

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Stockholm, SWE
88000 Employees

Ericsson Logo Ericsson

Master thesis; Securing AI-Generated Dependencies

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Stockholm, SWE
88000 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account