A Single-Loop Bilevel Deep Learning Method for Optimal Control of Obstacle Problems

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Song, Yongcun, Zeng, Shangzhi, Zhang, Jin, Zhang, Lvgang
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918276684054528
author Song, Yongcun
Zeng, Shangzhi
Zhang, Jin
Zhang, Lvgang
author_facet Song, Yongcun
Zeng, Shangzhi
Zhang, Jin
Zhang, Lvgang
contents Optimal control of obstacle problems arises in a wide range of applications and is computationally challenging due to its nonsmoothness, nonlinearity, and bilevel structure. Classical numerical approaches rely on mesh-based discretization and typically require solving a sequence of costly subproblems. In this work, we propose a single-loop bilevel deep learning method, which is mesh-free, scalable to high-dimensional and complex domains, and avoids repeated solution of discretized subproblems. The method employs constraint-embedding neural networks to approximate the state and control and preserves the bilevel structure. To train the neural networks efficiently, we propose a Single-Loop Stochastic First-Order Bilevel Algorithm (S2-FOBA), which eliminates nested optimization and does not rely on restrictive lower-level uniqueness assumptions. We analyze the convergence behavior of S2-FOBA under mild assumptions. Numerical experiments on benchmark examples, including distributed and obstacle control problems with regular and irregular obstacles on complex domains, demonstrate that the proposed method achieves satisfactory accuracy while reducing computational cost compared to classical numerical methods.
format Preprint
id arxiv_https___arxiv_org_abs_2601_04120
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle A Single-Loop Bilevel Deep Learning Method for Optimal Control of Obstacle Problems
Song, Yongcun
Zeng, Shangzhi
Zhang, Jin
Zhang, Lvgang
Optimization and Control
Machine Learning
68T07, 49M41, 65K15, 93-08
Optimal control of obstacle problems arises in a wide range of applications and is computationally challenging due to its nonsmoothness, nonlinearity, and bilevel structure. Classical numerical approaches rely on mesh-based discretization and typically require solving a sequence of costly subproblems. In this work, we propose a single-loop bilevel deep learning method, which is mesh-free, scalable to high-dimensional and complex domains, and avoids repeated solution of discretized subproblems. The method employs constraint-embedding neural networks to approximate the state and control and preserves the bilevel structure. To train the neural networks efficiently, we propose a Single-Loop Stochastic First-Order Bilevel Algorithm (S2-FOBA), which eliminates nested optimization and does not rely on restrictive lower-level uniqueness assumptions. We analyze the convergence behavior of S2-FOBA under mild assumptions. Numerical experiments on benchmark examples, including distributed and obstacle control problems with regular and irregular obstacles on complex domains, demonstrate that the proposed method achieves satisfactory accuracy while reducing computational cost compared to classical numerical methods.
title A Single-Loop Bilevel Deep Learning Method for Optimal Control of Obstacle Problems
topic Optimization and Control
Machine Learning
68T07, 49M41, 65K15, 93-08
url https://arxiv.org/abs/2601.04120