Skip to article sections
Purpose

This study aims to develop a data-driven Deep Koopman-based model predictive control (DK-MPC) framework for real-time trajectory tracking and stabilization of quadrotor systems. The objective is to overcome the computational limitations of conventional nonlinear model predictive control (NMPC) while preserving high-accuracy control performance.

Design/methodology/approach

A deep neural network-based Koopman operator is employed to map the nonlinear quadrotor dynamics into a globally linear latent space, enabling the formulation of a computationally efficient linear model predictive control (MPC) problem. The Koopman embeddings are trained using the publicly available WaveLab Pelican dataset to capture the coupled and nonlinear dynamics of the quadrotor. The proposed DK-MPC framework is evaluated through numerical simulations involving point stabilization and trajectory tracking tasks, including previously unseen helical trajectories. Performance is assessed in terms of tracking accuracy, computational efficiency and real-time feasibility, with comparisons made against conventional NMPC.

Findings

Simulation results demonstrate that the proposed DK-MPC framework achieves high-precision control with a coefficient of determination (R2) of 99%. The method requires approximately 10% of the computation time per control step compared to NMPC. For a prediction horizon of H = 50, DK-MPC maintains real-time feasibility with an average computation time of 15 ms per control step, while NMPC frequently fails to converge within the required time constraints. These findings confirm the effectiveness of the Koopman-based linear embedding in reducing computational burden without compromising control accuracy.

Originality/value

The study presents a scalable and computationally efficient integration of deep Koopman operator learning with MPC for quadrotor systems. By combining data-driven nonlinear system representation with linear MPC optimization, the proposed framework bridges the gap between advanced learning-based modeling and real-time predictive control implementation.

Quadrotor aerial robots, with their lightweight design and agile six-degree-of-freedom motion, have emerged as highly versatile platforms for research and practical applications [1, 2]. Their ability to perform rapid maneuvers, hover with precision and operate in constrained environments makes them suitable for diverse tasks such as aerial surveillance [3], package delivery [4], environmental monitoring [5] and search-and-rescue missions [6]. Furthermore, quadrotors are widely used as testbeds for advanced control strategies and embodied intelligence [7]. However, achieving reliable and high-performance control remains a challenge due to the strongly nonlinear dynamics of quadrotor flight, the tight coupling between translational and rotational states and the sensitivity to disturbances such as wind gusts and payload variations. These factors make precise modeling and control design difficult when relying solely on traditional analytical approaches [8–10].

To tackle these challenges, two dominant control methodologies have emerged in quadrotor research: model-based control and learning-based control [11]. Model-based strategies often employ simplified dynamics, such as linearization around hover conditions using Linear Quadratic Regulator (LQR) or feedback linearization techniques, to effectively approximate the quadrotor's behavior under specific assumptions [7, 12]. While these approaches offer structured frameworks for control design, their reliance on simplifying assumptions can lead to model-plant mismatches and suboptimal performance when faced with real-world disturbances or agile maneuvers [13].

To overcome the limitations of traditional model-based control, modern research has increasingly turned to learning-based strategies that leverage deep learning methods to better capture complex quadrotor dynamics and couplings [14]. One promising approach involves learning inverse dynamics: neural networks trained to map desired trajectories directly to actuator commands, simplifying controller implementation and enabling agile flight in challenging scenarios [15, 16]. However, the inherently underactuated nature of quadrotors and the possibility of multiple valid inverse mappings introduce challenges for these methods, potentially affecting generalization and stability.

Another avenue explored is reinforcement learning (RL) [17], where control policies are trained in simulation, either model-free or with learned models, and then deployed on real quadrotors [18]. While RL offers the flexibility to discover effective control strategies through experience, its deployment is often hindered by the sim-to-real gap. Transfers from simulation to real-world quadrotors can falter unless substantial domain randomization, real-world adaptation or robust policy learning techniques are applied [19].

To overcome the difficulties of modeling forward dynamics in quadrotor systems, Koopman operator–based methods have gained momentum, offering a compelling strategy for control design [20–22]. By embedding the nonlinear dynamics into a lifted space where the system evolves linearly, these approaches enable the deployment of efficient linear control techniques. However, a critical challenge lies in defining appropriate lifting (observable) functions; insufficient or suboptimal choices can introduce substantial modeling errors, negatively impacting prediction fidelity and controller effectiveness [23, 24].

While Koopman operator theory has recently gained traction for unmanned aerial vehicle (UAV) modeling and control, most existing studies rely on linear or shallow lifting architectures. These approaches often struggle to capture the strong nonlinearities inherent in quadrotor dynamics – particularly during high-rate or aggressive maneuvers – leading to reduced predictive fidelity and suboptimal closed-loop performance. To address this gap, we introduce a deep Koopman-based model predictive control (DK-MPC) framework that employs deep neural networks (DNNs) to automatically learn high-dimensional nonlinear embeddings. By achieving a more expressive, globally linearized representation of quadrotor dynamics, our method enables real-time optimization of control inputs. The resulting framework provides a scalable, data-driven alternative to conventional lifting methods, ensuring high-precision trajectory tracking and robust performance on physical quadrotor platforms.

The proposed Koopman-based approach belongs to a broader class of strategies aimed at nonlinear-to-linear conversion, most notably sharing conceptual goals with differential flatness. While both paradigms leverage linear structures to simplify controller design, they differ fundamentally in their mathematical foundations and implementation trade-offs. Quadrotors are well known to be differentially flat, meaning their states x and inputs u can be mapped to a set of flat outputs σ = [x, y, z, ψ]T and their finite derivatives. As highlighted by Li et al. [25], leveraging this property allows for exact linearization without the need for small-angle approximations, providing a rigorous framework for flatness-based model predictive control (MPC). However, while flatness provides an elegant analytical mapping, it typically relies on a precise nominal model and can struggle to account for unmodeled aerodynamic effects or complex ground-effect dynamics.

In contrast, Koopman-based methods utilize data-driven spectral lifting to approximate the nonlinear flight dynamics in a high-dimensional linear space. While differential flatness offers exactness within the constraints of the analytical model, Koopman operators provide a more flexible framework for capturing non-conservative forces or structural uncertainties that lack a simple flat representation. Thus, the choice between these methods involves a trade-off between the structural requirements of analytical flatness and the global, yet approximate, flexibility of data-driven lifting.

Numerical simulation experiments on a data-driven quadrotor model validate the effectiveness of the proposed DK-MPC framework. The results show that DK-MPC achieves highly accurate control performance, with clear improvements over conventional nonlinear control strategies. These outcomes highlight the potential of DK-MPC for future quadrotor applications, particularly in scenarios requiring precise trajectory tracking and robustness against dynamic variations [20]. Moreover, as a data-driven method, DK-MPC can be easily adapted to different quadrotor platforms, providing a flexible and scalable solution for advanced aerial robotics.

The remainder of this paper is organized as follows: Section 2 presents the development of the proposed DK-MPC framework, highlighting the integration of the deep Koopman operator with MPC. Section 3 reports the results of numerical simulation experiments, demonstrating the effectiveness of DK-MPC in enabling the quadrotor to accurately track reference trajectories while respecting system constraints. Finally, Section 4 concludes the paper with a summary of key findings and outlines directions for future research.

This section presents a comprehensive overview of the methodology behind the proposed DK-MPC approach, tailored to tackle the control complexities of quadrotors. It begins with an introduction to the overall architecture, followed by an explanation of the Koopman operator's formulation and learning process in high-dimensional settings, and concludes with the integration of the learned Koopman model into the MPC framework for control implementation.

Figure 1 depicts the structure of the proposed DK-MPC approach. It utilizes a deep learning-based Koopman operator to create a globally linear, time-invariant representation of the nonlinear quadrotor system by mapping the original state space into a higher-dimensional latent space. During each control cycle, the deep Koopman encoder maps the reference state xref into its corresponding high-dimensional latent representation. Simultaneously, the current state x, obtained from the quadrotor, is encoded into the same latent space. Within this transformed space, an MPC algorithm optimizes a quadratic cost function based on the latent states z and zref over a finite prediction horizon H, thereby producing the optimal control input u* to follow the reference trajectory.

Having a linear system representation in the latent space is essential because it enables the use of efficient quadratic programming solvers within the MPC framework. For nonlinear dynamics, the corresponding NMPC problem typically leads to a non-convex optimization that is computationally expensive and may fail to converge in real time, especially for fast systems such as quadrotors. By contrast, the Koopman-based linearization ensures that the MPC optimization remains convex and tractable, allowing control inputs to be computed within strict timing constraints while still capturing the essential nonlinear behavior of the original system. This balance between fidelity and computational efficiency is what makes the proposed DK-MPC framework suitable for real-time quadrotor control.

We examine a discrete-time nonlinear system governed by a function f, which describes the evolution of the state in the quadrotor. In this context, xk = [x, y, z, ϕ, θ, ψ]T denotes the system's state at time step k, representing the quadrotor's position and Euler angles, i.e. roll, pitch and yaw, respectively (Figure 2), and ukR4 represents the corresponding commanded motor speeds, such that:

(1)

Due to the inherent complexity and nonlinearity of f, directly designing a control strategy can be challenging. To address this, we apply the Koopman operator framework, which operates in a transformed (lifted) space defined by a mapping function ϕ. This function lifts the original nonlinear system into a higher-dimensional space where its behavior becomes approximately linear:

(2)

The effectiveness of this linear approximation relies heavily on the choice of the lifting function ϕ. In this study, we adopt a deep learning-based method to learn ϕ enabling us to model the complex nonlinear dynamics typical of quadrotors.

Extending the Koopman operator framework, the lifting function ϕ(x, u) is separated into two components: one dependent on the state, ϕx(x), and the other on the input, ϕu(u), such that:

(3)

This decomposition enables the system dynamics to be expressed in a control-affine form through matrix operations:

(4)

From this, the evolution of the lifted state can be written as:

(5)

Consistent with simplifications proposed in earlier studies [26, 27], the input lifting function ϕu(u) is directly represented by the control input u. Defining Kxx = A, Kxu = B and ϕx(xk) = zk, the dynamics simplify to a globally linear form:

(6)

This transformation yields a linear state space model that not only facilitates controller synthesis but also effectively approximates the overall behavior of the original nonlinear quadrotor system.

This subsection outlines the deep learning strategy employed to estimate the Koopman operator and its associated lifting and inverse functions. As depicted in Figure 3, the method is based on a deep auto-encoder architecture designed to learn the forward mapping ϕ(θ) and its inverse ϕ(θ)−1. The auto-encoder comprises two primary components: an encoder and a decoder. The encoder ϕ(θ), implemented using a multi-layer perceptron (MLP), lifts the system's original state into a higher-dimensional latent space, where the system dynamics become approximately linear. Conversely, the decoder ϕ(θ)−1 reconstructs the original state from this latent representation. Although alternative architectures such as convolutional neural networks or recurrent neural networks could also be promising for capturing spatial or temporal correlations in the data, the MLP was chosen for its simplicity, general approximation capability and efficiency in training on tabular state-action datasets such as those considered in this work.

To model the system's linear behavior in the latent space, two additional single-layer MLPs are employed to approximate the matrices A and B from Equation (6). These networks are designed without biases or activation functions, enabling them to function as straightforward linear transformations. This structure ensures that the learned latent dynamics remain consistent with the theoretical assumptions of the Koopman framework.

Reconstruction loss: To accurately model the relationship between the original and latent state representations, a reconstruction loss is defined using the L2 norm. This loss ensures that the auto-encoder learns to map states to and from the latent space with minimal error:

(7)

Linear dynamics loss: To identify the linear operator A and the control matrix B, we define a dynamics loss that encourages the latent state transitions to follow a linear model structure. This objective penalizes deviations from the expected linear behavior in the latent space:

(8)

Stability loss: To promote stability in the learned Koopman operator A, we introduce a stability loss that penalizes cases where the spectral radius exceeds 1. Specifically, we compute the spectral radius ρ(A) as the maximum absolute eigenvalue of A and apply a rectified linear unit-based penalty to constrain it within the unit circle:

(9)

This loss term helps ensure the linear dynamics model remains stable by discouraging eigenvalues with magnitudes greater than one.

Total loss function: The final objective, as shown in Equation (10), is expressed as a weighted sum of the reconstruction loss, linear dynamics loss, stability loss, along with an L2 regularization term to the model weights W to reduce overfitting [28]:

(10)

This integrated framework supports end-to-end learning of the Koopman-based model by jointly training the linear operator A, the control-affine matrix B and the lifting function ϕ. Consequently, it enables effective modeling and control of the nonlinear dynamics inherent in quadrotor systems.

Once the Koopman operator has been learned, we employ MPC to regulate the nonlinear behavior of the quadrotor. MPC is particularly well suited for systems with constraints and excels in optimizing control trajectories by predicting future system evolution [16]. Utilizing the linear system representation obtained via the Koopman framework, we can formulate an MPC scheme that is both efficient and effective.

The controller solves an optimization problem to compute an optimal sequence of control actions uˆt:t+H* over a finite prediction horizon HN. The cost function penalizes both the deviation from the reference trajectory in the lifted space and the magnitude of the control inputs:

(11)

Here, QˆRn×n and RˆRm×m are positive semi-definite weighting matrices that penalize state tracking error and control effort, respectively. The variables umin and umax define the allowable range for the control inputs, while n denotes the dimension of the latent (lifted) state space. At each control timestep, we encode the current system state xt and future reference states xt:t+Href into the latent space, yielding zˆt and zˆt:t+Href. Solving the optimization problem in Equation (11) produces an optimal control sequence, from which only the first control action uˆt* is applied to the system. This process is repeated iteratively until convergence is achieved.

This section outlines the methodology for data preparation and the training procedure for the proposed model. We then deploy the DK-MPC controller to manage the motion of the quadrotor simulated using the system dynamics of equations depicted in Reference [29]. To validate the effectiveness of the proposed approach, we conduct three experimental scenarios: a path-following task, a moving target tracking task and a complex helical trajectory tracking task. The first experiment highlights the precision and fast dynamic response achieved by DK-MPC; the second illustrates its capability to adapt to dynamic conditions, while the third assesses the responsiveness of the control system to handle complex trajectories, indicating its potential for real-world deployment in responsive control applications.

To evaluate the proposed data-driven DK-MPC strategy in a dynamic aerial platform, we employ the dataset of the AscTec Pelican quadrotor, a research-grade UAV known for its modularity and robust performance. The quadrotor features a rigid carbon-fiber frame with a diagonal motor-to-motor span of approximately 85 cm and a total weight of around 1.6 kg, including battery and onboard electronics. It is powered by four brushless direct current motors equipped with fixed-pitch propellers and is capable of both high-thrust maneuvers and stable hovering.

The Pelican is equipped with an onboard computer running a real-time operating system, supporting integration with Robot Operating System (ROS) for high-level control and sensor fusion. It includes an inertial measurement unit, barometer and global positioning system module for autonomous navigation and offers payload options for additional sensors or manipulators. The platform's flexibility and open software interface make it ideal for implementing and validating advanced control algorithms, including the proposed DK-MPC framework.

For training and evaluating the proposed DK-MPC framework, we utilize the publicly available Pelican Dataset provided by the WaveLab group [29, 30]. This dataset was collected using the AscTec Pelican quadrotor in a controlled indoor environment equipped with a motion capture system. The dataset includes time-series recordings of the quadrotor's state and control inputs across various flight maneuvers.

Specifically, the dataset contains 35 different flights of measurements such as position, velocity, acceleration, orientation (quaternions), angular rates and motor commands, all sampled at 100 Hz. These high-fidelity recordings offer a comprehensive representation of the quadrotor's dynamic behavior and are well suited for learning both the lifting functions and the Koopman operator used in the DK-MPC framework.

To ensure robust training, we preprocess the dataset by segmenting it into input-output pairs (xk, uk, xk+1), normalizing each feature and partitioning the data into training and validation sets. The richness and quality of this dataset make it ideal for modeling complex nonlinear dynamics in aerial robotics.

To facilitate robust learning, the dataset was divided into three subsets: training, validation and testing. Prior to training, all input features were normalized using the min-max scaling approach [31], which linearly transforms each feature to the range [−1, 1]. This normalization improves convergence and maintains consistency throughout the training process. The transformation is defined by:

(12)

This preprocessing step ensures that all features contribute proportionally during optimization, preventing dominance by features with larger numerical ranges [32].

The hyperparameters employed for training the proposed architecture are listed in Table 1. These were carefully selected to achieve a balance between model accuracy and training stability and were tuned through empirical testing and validation performance.

To evaluate the learning behavior of the proposed DK-MPC framework, we monitored the evolution of various loss components during training, as shown in Figure 4. The plot illustrates the convergence of the linear dynamics loss, reconstruction loss, stability loss and the total loss over 50 epochs. All losses exhibit a rapid decline within the first 10 epochs, stabilizing at near-zero values, which indicates fast convergence and training stability. The smooth decrease in the total loss suggests that the model effectively learns a Koopman-consistent latent representation, simultaneously satisfying reconstruction accuracy and spectral radius constraints. This behavior confirms the effectiveness of the loss design in guiding the model toward a stable and accurate linear approximation of the nonlinear system dynamics.

To evaluate the prediction accuracy of the learned Koopman-based model, we assess its one-step-ahead forecasting performance by comparing the predicted next state xk+1 with the ground-truth test data across multiple state dimensions, including position (x, y, z) and roll angle (ϕ), as illustrated in Figure 5. The model's predictions (orange) closely match the true states (blue dashed lines), demonstrating its ability to reliably estimate xk+1 from the current state-input pair (xk, uk). This strong agreement across all state channels indicates that the learned lifted linear representation is sufficiently expressive to capture the essential nonlinear dynamics of the quadrotor. These results confirm the suitability of the Koopman framework for accurate next-state prediction and support its effectiveness as a foundation for data-driven control design in complex dynamical systems.

To determine the appropriate size of the Koopman latent space, we conducted a sensitivity analysis by evaluating latent dimensions of 4, 6, 8, 12 and 16 against the prediction accuracy and the computation time, including the complete online control pipeline. The results, summarized in Table 2, show a clear trade-off between model accuracy and computational performance. Lower-dimensional embeddings (4 and 6) led to under-expressive representations, achieving average R2 scores of only 71.32% and 86.13%, respectively. Increasing the latent dimension to 8 yielded a substantial improvement, achieving 99.45% prediction accuracy while maintaining a low computation time of 0.008 s per control step. Higher dimensions (12 and 16) offered only marginal gains in accuracy (up to 99.57%) but came with increased computational cost. Based on this trade-off, a latent dimension of 8 was selected as the optimal balance between model expressiveness and real-time feasibility for DK-MPC.

For a fair and transparent comparison with the proposed Koopman-based predictive controller, a conventional nonlinear MPC (NMPC) baseline was implemented using the full dynamic model of the quadrotor in DK-MPC. The NMPC formulation incorporated the system states for position (x, y, z), linear velocities, Euler angles and angular rates. Motor thrusts were represented through nonlinear actuator maps using the constants bi and ki, where each rotor force was modeled as fi=(ui2+bi)/ki, thus capturing the nonlinearities inherent in quadrotor actuation. Physical parameters, including the arm length l = 0.2 m and drag constant d = 0.02, were embedded directly within the dynamic equations to ensure accurate prediction as indicated in Appendix A. The controller was solved using the default DK-MPC nonlinear optimizer with a single-shooting approach and a discretization consistent with the simulation step of 0.05 s. In the NMPC, the weighting matrices have been chosen to be Q = Diag([100, 100, 100, 10]) and R = Diag([0.001, 0.001, 0.001, 0.001]). The NMPC utilizes the dynamic position and rotation model alongside the quadrotor parameters depicted in Ref. [29] as the MPC prediction model.

In the point stabilization experiment, the performance of DK-MPC is evaluated against a conventional NMPC with the same prediction horizon H across stepwise changes in target states. As shown in Figure 6, the DK-MPC controller demonstrates superior tracking accuracy and stability across all state dimensions. It responds quickly to abrupt changes in reference positions (x, y and z) and orientation (ϕ), with minimal overshoot and short settling time.

In contrast, the NMPC exhibits noticeable oscillations, slower convergence and greater steady-state error – particularly in the rotational dynamics. These results validate the capability of the learned Koopman-based linear representation to generalize well to unseen reference inputs, ensuring robust and efficient control in point-to-point tasks. The effectiveness of DK-MPC in handling sharp transitions also highlights its potential for real-time applications in agile and dynamic environments.

To rigorously assess the stability of the proposed DK-MPC framework, we analyze the spectral properties of the learned discrete-time Koopman transition matrix, denoted as ARN×N. In this data-driven linear representation, the autonomous evolution of the lifted state zk is governed by the linear relation zk+1 = Azk.

According to discrete-time control theory, the global stability of the equilibrium point is determined by the eigenvalues λi of the matrix A. The system is considered Lyapunov stable if all eigenvalues lie within the unit circle in the complex z-plane:

(13)

Figure 7 illustrates the eigenvalue distribution derived from our trained model. The results show that the eigenvalues are strictly contained within the unit circle (|λi| < 1) or reside at the marginal stability point (1, 0). The poles at (1, 0) correspond to the physical integrators of the quadrotor system (e.g. global position coordinates x, y, z), while the remaining poles, associated with the DNN embeddings, are strictly dissipative.

This spectral distribution provides a formal guarantee that the learned nonlinear embeddings do not introduce divergent modes. Consequently, the DK-MPC optimizer operates on a stable linear manifold, ensuring that the predicted trajectories remain bounded over the prediction horizon Hp.

In the trajectory tracking experiment, the DK-MPC is further evaluated in a continuous motion scenario where the reference path evolves smoothly over time. As illustrated in Figure 8, DK-MPC consistently outperforms the baseline NMPC across all translational (x, y, z) and rotational (ϕ) states with the same prediction horizon H. The DK-MPC controller closely follows the reference trajectory with reduced tracking error and smoother control responses, while the NMPC exhibits lagging behavior and higher deviation, particularly during transitions and changes in trajectory curvature. These observations emphasize the effectiveness of the Koopman-based linear model in capturing global dynamics, allowing the controller to anticipate and adjust to trajectory changes more efficiently. The results affirm that DK-MPC not only achieves high tracking accuracy but also maintains stability and robustness during dynamic flight tasks.

To further evaluate the performance and scalability of the controllers for real-time applications, we analyze how both DK-MPC and NMPC behave across varying prediction horizon lengths. As illustrated in Figure 9, DK-MPC consistently achieves high R2 scores above 0.99 across all horizons, demonstrating robust prediction accuracy and reliable model generalization. In contrast, the NMPC shows extremely poor accuracy for shorter horizons (H = 5, 10 and 15), with significantly negative R2 values, indicating divergence from the reference trajectory. Only at longer horizons (H = 20 and 25) does it begin to exhibit modest improvement.

On the computational side, DK-MPC maintains a low and stable control time below 5 ms per step regardless of horizon length, highlighting its suitability for real-time applications. NMPC, however, shows a steep increase in computational cost, reaching nearly 60 ms at H = 20. These results underscore the advantage of DK-MPC in achieving accurate control with minimal computational overhead, especially when scalability and fast response are critical in embedded or onboard systems.

The helical trajectory was chosen as a benchmark task because it simultaneously excites both the translational and rotational dynamics of the quadrotor while requiring continuous changes in altitude, heading and lateral position. Such a trajectory is highly nonlinear, combines periodic and ascending motion and is more challenging than standard step or straight-line paths. This makes it a suitable stress test for evaluating the ability of predictive controllers to handle coupled dynamics and stringent real-time constraints.

An important observation from the helical trajectory tracking experiment is that the conventional NMPC formulation was unable to find feasible control solutions to follow the desired reference. The strong nonlinearities of the quadrotor dynamics, combined with the complexity of the helical reference path, led to optimization problems that either diverged or failed to converge within the required real-time constraints. As a result, NMPC could not successfully generate control inputs that ensured stable trajectory tracking. In contrast, the proposed DK-MPC framework maintained computational tractability by leveraging a globally linearized Koopman-based model, which enabled efficient optimization at each step as depicted in Figure 10. This distinction underscores the robustness and practicality of the Koopman-based approach for solving constrained trajectory tracking problems in real time, particularly where traditional NMPC struggles to deliver feasible solutions.

While the proposed DK-MPC framework is developed and validated in simulation, its structure is well suited to address real-world challenges that extend beyond the training dataset. In practical deployment scenarios, quadrotors are subject to external disturbances such as wind gusts, parameter drift and sensor noise, all of which may degrade controller performance if not anticipated during the design process. The deep Koopman model provides an advantageous structure for handling such uncertainties, as its learned latent space representation can be enhanced through disturbance-augmented training or domain randomization techniques. Incorporating these variations during training enables the learned Koopman operator to generalize beyond nominal operating conditions and maintain prediction fidelity in the presence of unmodeled effects.

Additionally, the MPC layer inherently supports robustness through constraint tightening, disturbance-aware prediction or tube-based formulations. Such strategies can be integrated without modifying the Koopman representation and allow the controller to remain stable under bounded disturbances. Sensor noise can be mitigated through filtering techniques or by fusing the Koopman model with state estimators, such as extended Kalman filters or Koopman-based observers, further improving reliability in real-world operation. Together, these considerations highlight the adaptability of the DK-MPC approach and its potential to maintain performance when deployed on physical quadrotor platforms exposed to environmental and measurement uncertainties.

In this work, we introduced a DK-MPC framework for quadrotor trajectory tracking and stabilization. By embedding nonlinear dynamics into a lifted linear space via DNNs, DK-MPC enabled computationally efficient MPC formulations that outperformed traditional NMPC in both accuracy and real-time feasibility. Simulation results on trajectory tracking and point stabilization tasks confirmed that DK-MPC significantly reduced tracking errors and achieved average control step times of 15 ms for H = 50, whereas NMPC often failed to provide feasible solutions within real-time constraints. The study highlights three main contributions: (1) the development of a black-box Koopman embedding for quadrotor dynamics trained on flight data; (2) integration of this model with MPC for constrained real-time control and (3) extensive validation against NMPC on challenging trajectories.

The computational efficiency of the DK-MPC approach enables capabilities that are often impractical for conventional NMPC due to real-time computational constraints. In particular, we note that the reduced-order linear structure of the Koopman-based predictor allows faster optimization cycles, which can support higher-frequency control loops, onboard deployment on resource-limited processors and more complex mission objectives requiring long-horizon reasoning. This efficiency also opens the door to advanced functionalities – such as dynamic obstacle avoidance, real-time trajectory re-planning, multi-agent coordination and adaptive control under changing dynamics – that typically exceed the computational budget of full NMPC. By reducing the computational load while preserving nonlinear modeling fidelity, the proposed framework represents a promising step toward enabling more sophisticated autonomous behaviors on real aerial robotic platforms.

Future research will focus on extending DK-MPC towards online or adaptive Koopman learning for robustness under time-varying dynamics and disturbances, deploying the controller on embedded flight hardware to validate on-board real-time performance and scaling the approach to cooperative multi-quadrotor systems. These directions aim to further demonstrate the potential of Koopman-based MPC as a unifying framework for reliable and efficient control in advanced aerial robotics. Meanwhile, future research will extend the Koopman-based predictive control framework to multi-quadrotor systems operating in coordinated formations or platoons [33–35]. In such scenarios, inter-vehicle dynamics introduce additional nonlinear coupling effects that may significantly influence stability, robustness and tracking accuracy.

The supplementary material for this article can be found online.

1.
Foehn
 
P
,
Kaufmann
 
E
,
Romero
 
A
,
Penicka
 
R
,
Sun
 
S
,
Bauersfeld
 
L
,
Laengle
 
T
,
Cioffi
 
G
,
Song
 
Y
,
Loquercio
 
A
,
Scaramuzza
 
D
.
Agilicious: open-source and open-hardware agile quadrotor for vision-based flight
.
Sci Robot
.
2022
;
7
(
67
): eabl6259. doi: .
2.
Zulu
 
A
,
John
 
S
.
A review of control algorithms for autonomous quadrotors
.
Open J Appl Sci
.
2014
;
4
(
14
):
547
-
56
. doi: .
3.
Aizelman
 
I
,
Magazinnik
 
D
,
Feldman
 
D
,
Klein
 
I
.
Quadrotor with wheels: design and experimental evaluation
.
Sci Rep
.
2024
;
14
(
1
): 15603. doi: .
4.
Saunders
 
J
,
Saeedi
 
S
,
Li
 
W
.
Autonomous aerial robotics for package delivery: a technical review
.
J Field Robot
.
2024
;
41
(
1
):
3
-
49
. doi: .
5.
Kokate
 
P
,
Middey
 
A
,
Sadistap
 
S
,
Sarode
 
G
,
Narayan
 
A
.
Review on drone-assisted air-quality monitoring systems
.
Drones Autonomous Vehicles
.
2023
;
1
(
1
): 10005. doi: .
6.
Lyu
 
M
,
Zhao
 
Y
,
Huang
 
C
,
Huang
 
H
.
Unmanned aerial vehicles for search and rescue: a survey
.
Remote Sens
.
2023
;
15
(
13
):
3266
. doi: .
7.
Khalid
 
A
,
Mushtaq
 
Z
,
Arif
 
S
,
Zeb
 
K
,
Khan
 
MA
,
Bakshi
 
S
.
Control schemes for quadrotor UAV: taxonomy and survey
.
ACM Comput Surv
.
2023
;
56
(
5
):
1
-
32
. doi: .
8.
Dhadekar
 
DD
,
Sanghani
 
PD
,
Mangrulkar
 
K
,
Talole
 
S
.
Robust control of quadrotor using uncertainty and disturbance estimation
.
J Intell Robot Syst
.
2021
;
101
(
3
):
60
. doi: .
9.
Hanover
 
D
,
Foehn
 
P
,
Sun
 
S
,
Kaufmann
 
E
,
Scaramuzza
 
D
.
Performance, precision, and payloads: adaptive nonlinear MPC for quadrotors
.
IEEE Robot Automation Lett
.
2021
;
7
(
2
):
690
-
7
. doi: .
10.
Wang
 
C
,
Song
 
B
,
Huang
 
P
,
Tang
 
C
.
Trajectory tracking control for quadrotor robot subject to payload variation and wind gust disturbance
.
J Intell Robot Syst
.
2016
;
83
(
2
):
315
-
33
. doi: .
11.
Sönmez
 
S
,
Rutherford
 
MJ
,
Valavanis
 
KP
.
A survey of offline-and online-learning-based algorithms for multirotor uavs
.
Drones
.
2024
;
8
(
4
):
116
. doi: .
12.
Belkheiri
 
M
,
Rabhi
 
A
,
El Hajjaji
 
A
,
Pegard
 
C
.
Different linearization control techniques for a quadrotor system
. In:
CCCA12
.
IEEE
;
2012
. p.
1
-
6
.
13.
Salzmann
 
T
,
Kaufmann
 
E
,
Arrizabalaga
 
J
,
Pavone
 
M
,
Scaramuzza
 
D
,
Ryll
 
M
.
Real-time neural MPC: deep learning model predictive control for quadrotors and agile robotic platforms
.
IEEE Robot Automation Lett
.
2023
;
8
(
4
):
2397
-
404
. doi: .
14.
Saviolo
 
A
,
Li
 
G
,
Loianno
 
G
.
Physics-inspired temporal learning of quadrotor dynamics for accurate model predictive trajectory tracking
. ,
2022
.
15.
Zhou
 
S
,
Helwa
 
MK
,
Schoellig
 
AP
.
An inversion-based learning approach for improving control in quadrotor systems
.
IEEE Robot Automation Lett
.
2018
;
3
(
3
):
2366
-
73
.
16.
El-Hussieny
 
H
.
Real-time deep learning-based model predictive control of a 3-dof biped robot leg
.
Sci Rep
.
2024
;
14
(
1
): 16243. doi: .
17.
El-Hussieny
 
H
,
Hameed
 
IA
.
Obstacle-aware navigation of soft growing robots via deep reinforcement learning
.
IEEE Access
.
2024
;
12
:
38192
-
201
. doi: .
18.
Hwangbo
 
J
,
Sa
 
I
,
Siegwart
 
R
,
Hutter
 
M
.
Control of a quadrotor with reinforcement learning
.
IEEE Robot Automation Lett
.
2017
;
2
(
4
):
2096
-
103
. doi: .
19.
Dionigi
 
A
,
Sun
 
H
,
Omerdic
 
E
,
Floreano
 
D
,
Carrillo
 
L
.
The power of input: benchmarking zero-shot sim-to-real transfer of reinforcement learning control policies for quadrotor control
. ,
2024
.
20.
Narayanan
 
SSKS.
,
Tellez-Castro
 
D
,
Sutavani
 
S
,
Vaidya
 
U
.
SE(3) Koopman-MPC: data-driven learning and control of quadrotor UAVs
.
IFAC-PapersOnLine
.
2023
;
56
(
3
):
607
-
12
. doi: .
21.
Abido
 
MA
,
Manaa
 
ZM
,
Abdallah
 
AM
,
Ali
 
SSA
.
Koopman-LQR controller for quadrotor UAVs from data
. ,
2024
.
22.
Martini
 
S
.
Koopman-based modeling for nonlinear control of multirotor UAVs
.
PhD thesis, University of Denver
,
2024
.
23.
Rajkumar
 
SM
,
Cheng
 
S
,
Hovakimyan
 
N
,
Goswami
 
D
.
Linear model predictive control for quadrotors with an analytically derived Koopman model
. ,
2024
.
24.
Shi
 
L
,
Haseli
 
M
,
Mamakoukas
 
G
,
Bruder
 
D
,
Abraham
 
I
,
Murphey
 
T
,
Cortés
 
J
,
Karydis
 
K
.
Koopman operators in robot learning: theory, algorithms, and applications
. ,
2024
.
25.
Li
 
Y
,
Zhu
 
Q
,
Elahi
 
A
.
Quadcopter trajectory tracking control based on flatness model predictive control and neural network
.
Actuators
.
2024
;
13
(
4
):
154
. doi: .
26.
Bruder
 
D
,
Fu
 
X
,
Gillespie
 
RB
,
Remy
 
CD
,
Vasudevan
 
R
.
Data-driven control of soft robots using koopman operator theory
.
IEEE Trans Robot
.
2020
;
37
(
3
):
948
-
61
. doi: .
27.
Shi
 
H
,
Meng
 
MQ-H
.
Deep koopman operator with control for nonlinear systems
.
IEEE Robot Automation Lett
.
2022
;
7
(
3
):
7700
-
7
. doi: .
28.
Ng
 
AY
.
Feature selection, l 1 vs. l 2 regularization, and rotational invariance
. In:
Proceedings of the Twenty-First International Conference on Machine Learning
;
2004
. p.
78
.
29.
Mohajerin
 
N
,
Mozifian
 
M
,
Waslander
 
S
.
Deep learning a quadrotor dynamic model for multi-step prediction
. In:
2018 IEEE International Conference on Robotics and Automation (ICRA)
.
IEEE
;
2018
. p.
2454
-
9
.
30.
Mohajerin
 
N
,
Waslander
 
SL
.
Multistep prediction of dynamic systems with recurrent neural networks
.
IEEE Trans Neural Netw Learn Syst
.
2019
;
30
(
11
):
3370
-
83
. doi: .
31.
Kumar
 
M
,
Stoll
 
R
,
Stoll
 
N
.
A min-max approach to fuzzy clustering, estimation, and identification
.
IEEE Trans Fuzzy Syst
.
2006
;
14
(
2
):
248
-
62
. doi: .
32.
Abdelaziz
 
H
,
Ahmed
 
A
,
El-Hussieny
 
H
.
Approximate neural network-based nonlinear model predictive control of soft continuum robots
. In:
2024 20th IEEE/ASME International Conference on Mechatronic and Embedded Systems and Applications (MESA)
.
IEEE
;
2024
. p.
1
-
7
.
33.
Li
 
S
,
Chen
 
C
,
Zheng
 
H
,
Liu
 
Y
,
Xu
 
Q
,
Li
 
K
.
Nonlinear data-driven predictive control for mixed platoons based on Koopman operator
. In:
Proceedings of the 8th CAA International Conference on Vehicular Control and Intelligence (CVCI)
.
IEEE
;
2024
. p.
1
-
6
.
34.
Li
 
S
,
Wang
 
J
,
Yang
 
K
,
Xu
 
Q
,
Wang
 
J
,
Li
 
K
.
Robust nonlinear data-driven predictive control for mixed vehicle platoons via Koopman operator and reachability analysis
. ,
2025
.
35
Lyu
 
H
,
Guo
 
Y
,
Liu
 
P
,
Zheng
 
N
,
Wang
 
T
,
Yue
 
Q
.
Mitigating traffic oscillations in mixed traffic flow with scalable deep Koopman predictive control
.
Adv Eng Inform
.
2026
;
71
:
104258
-
104279
.
Published in Applied Computing and Informatics. Published by Emerald Publishing Limited. This article is published under the Creative Commons Attribution (CC BY 4.0) licence. Anyone may reproduce, distribute, translate and create derivative works of this article (for both commercial and non-commercial purposes), subject to full attribution to the original publication and authors. The full terms of this licence may be seen at Link to the terms of the CC BY 4.0 licence.

Supplementary data

Data & Figures

Figure 1
A control system diagram showing DL-Koopman MPC with encoders, lifted model blocks, and a quadrotor loop.On the far left, a rectangular block labeled “Encoder” is shown. A horizontal arrow labeled “x subscript r” enters this block from the left. Inside the encoder, a neural network pattern appears with multiple small circular nodes arranged in vertical layers and connected by diagonal lines, with colored vertical side panels. From this encoder, a horizontal arrow labeled “z subscript r” exits to the right and enters a large rounded rectangular block labeled “D L-Koopman M P C”. Within the “D L-Koopman M P C” block, a rectangular block labeled “Model Predictive Control (M P C)” is positioned near the bottom center. Above and slightly to the right, a rounded rectangular box labeled “Linear Lifted Model” is placed. Inside this lifted model, a circular summation node with a plus symbol is located toward the upper-left area. Two rectangular blocks labeled “A of (theta)” positioned to the right and “B of (theta)” positioned below are connected with solid arrows into the summation node. A horizontal arrow labeled “z subscript k” enters the lifted model from the right. From the summation node, a dashed arrow labeled “z subscript k plus 1” extends downward into the M P C block. Another dashed arrow labeled “u bar subscript k” extends upward from the M P C block into the “B of (theta)” block. From the bottom of the M P C block, a vertical arrow labeled “u asterisk” exits downward from the “D L-Koopman M P C” block to a rectangular image labeled “Asc Tec Pelican Quadrotor”, showing a central body with four extended arms arranged in a cross pattern. To the right of the quadrotor, a second rectangular block labeled “Encoder” is shown with a similar neural network pattern of layered circular nodes and connecting lines. A horizontal arrow labeled “x subscript k” enters this encoder from the left. From this encoder, a vertical line extends upward along the right side, curves across the top, and connects back into the lifted model at the input labeled “z subscript k”. Another downward arrow from the encoder loops back to the left and upward to the M P C block.

Schematic of the proposed DK-MPC framework for controlling the AscTec Pelcian quadrotor. The deep Koopman operator projects both the reference state xref and the current state x into a high-dimensional latent space where the system exhibits linear behavior. Using these latent representations and the corresponding linear dynamics, the MPC controller computes optimal control inputs u* to guide the quadrotor along the desired trajectory

Figure 1
A control system diagram showing DL-Koopman MPC with encoders, lifted model blocks, and a quadrotor loop.On the far left, a rectangular block labeled “Encoder” is shown. A horizontal arrow labeled “x subscript r” enters this block from the left. Inside the encoder, a neural network pattern appears with multiple small circular nodes arranged in vertical layers and connected by diagonal lines, with colored vertical side panels. From this encoder, a horizontal arrow labeled “z subscript r” exits to the right and enters a large rounded rectangular block labeled “D L-Koopman M P C”. Within the “D L-Koopman M P C” block, a rectangular block labeled “Model Predictive Control (M P C)” is positioned near the bottom center. Above and slightly to the right, a rounded rectangular box labeled “Linear Lifted Model” is placed. Inside this lifted model, a circular summation node with a plus symbol is located toward the upper-left area. Two rectangular blocks labeled “A of (theta)” positioned to the right and “B of (theta)” positioned below are connected with solid arrows into the summation node. A horizontal arrow labeled “z subscript k” enters the lifted model from the right. From the summation node, a dashed arrow labeled “z subscript k plus 1” extends downward into the M P C block. Another dashed arrow labeled “u bar subscript k” extends upward from the M P C block into the “B of (theta)” block. From the bottom of the M P C block, a vertical arrow labeled “u asterisk” exits downward from the “D L-Koopman M P C” block to a rectangular image labeled “Asc Tec Pelican Quadrotor”, showing a central body with four extended arms arranged in a cross pattern. To the right of the quadrotor, a second rectangular block labeled “Encoder” is shown with a similar neural network pattern of layered circular nodes and connecting lines. A horizontal arrow labeled “x subscript k” enters this encoder from the left. From this encoder, a vertical line extends upward along the right side, curves across the top, and connects back into the lifted model at the input labeled “z subscript k”. Another downward arrow from the encoder loops back to the left and upward to the M P C block.

Schematic of the proposed DK-MPC framework for controlling the AscTec Pelcian quadrotor. The deep Koopman operator projects both the reference state xref and the current state x into a high-dimensional latent space where the system exhibits linear behavior. Using these latent representations and the corresponding linear dynamics, the MPC controller computes optimal control inputs u* to guide the quadrotor along the desired trajectory

Close Figure 1
Figure 2
A quadrotor body and inertial frame diagram showing axes, rotors, and angles phi, theta, and psi.The central red circular hub is connected to four black circular nodes by straight black arms forming an “X” shape. The four outer nodes are labeled “1”, “2”, “3”, and “4”, each placed at the end of an arm. The node labeled “1” is positioned toward the lower-right, “2” toward the lower-left, “3” toward the upper-left, and “4” toward the upper-right. A label “Body Frame” appears near the central region. From the central hub, a red arrow extends diagonally upper left side, showing one axis of the body frame. Near this arrow, a counterclockwise curved arrow is drawn and labeled “psi”. Another red arrow extends diagonally toward the lower-right from the center, aligned with the arm pointing to node “1”, and is labeled “Front”. Near this arrow, a counterclockwise curved arrow is labeled “phi”. A third red arrow extends diagonally toward the lower-left, aligned near the arm pointing to node “2”, with a clockwise curved arrow labeled “theta”. At the lower-left of the diagram, a separate coordinate system labeled “Inertial Frame” is shown. Three arrows originate from a common point: one blue arrow pointing upward, one green arrow pointing diagonally upward to the right, and one red arrow pointing horizontally to the right. A dashed line extends from this inertial frame origin toward the central hub of the quadrotor. Each rotor node is surrounded by a faint circular shaded region. The black arms connect symmetrically from the central hub to each of the four labeled nodes, forming a cross-like pattern rotated diagonally.

Quadrotor frames and variables

Figure 2
A quadrotor body and inertial frame diagram showing axes, rotors, and angles phi, theta, and psi.The central red circular hub is connected to four black circular nodes by straight black arms forming an “X” shape. The four outer nodes are labeled “1”, “2”, “3”, and “4”, each placed at the end of an arm. The node labeled “1” is positioned toward the lower-right, “2” toward the lower-left, “3” toward the upper-left, and “4” toward the upper-right. A label “Body Frame” appears near the central region. From the central hub, a red arrow extends diagonally upper left side, showing one axis of the body frame. Near this arrow, a counterclockwise curved arrow is drawn and labeled “psi”. Another red arrow extends diagonally toward the lower-right from the center, aligned with the arm pointing to node “1”, and is labeled “Front”. Near this arrow, a counterclockwise curved arrow is labeled “phi”. A third red arrow extends diagonally toward the lower-left, aligned near the arm pointing to node “2”, with a clockwise curved arrow labeled “theta”. At the lower-left of the diagram, a separate coordinate system labeled “Inertial Frame” is shown. Three arrows originate from a common point: one blue arrow pointing upward, one green arrow pointing diagonally upward to the right, and one red arrow pointing horizontally to the right. A dashed line extends from this inertial frame origin toward the central hub of the quadrotor. Each rotor node is surrounded by a faint circular shaded region. The black arms connect symmetrically from the central hub to each of the four labeled nodes, forming a cross-like pattern rotated diagonally.

Quadrotor frames and variables

Close Figure 2
Figure 3
A neural network diagram with layered nodes, transformation blocks, and a summation node connected in sequence.On the far left, a vertical rectangular column labeled “x subscript k” contains three circular nodes arranged vertically inside a narrow red-outlined box. From this column, multiple diagonal lines fan out to the right into a larger rectangular block labeled “Encoder”, which is outlined in green. Inside the encoder, eight circular nodes are arranged vertically, with dense diagonal connections forming a layered neural network pattern. From the right side of the encoder, connections lead into a narrow vertical column of five circular nodes labeled “z subscript k”, enclosed in a light yellow box. From this column, a horizontal arrow extends to the right into a square block labeled “A of (theta subscript A)”, which contains a grid pattern of small squares. A line exits the right side of this block and curves downward toward a circular summation node marked with a plus symbol. Below the “z subscript k” column, a second narrow vertical column labeled “u subscript k” contains four circular nodes inside a blue-outlined box. A horizontal arrow extends from this column into another square block labeled “B of (theta subscript B)”, which also contains a grid pattern. A line exits this block and curves upward into the same circular summation node. From the summation node, a horizontal arrow extends to the right into another narrow vertical column labeled “z subscript k plus 1”, again shown as stacked five circular nodes inside a light yellow box. From this column, multiple diagonal connections spread into a second large green-outlined rectangular block labeled “Decoder”, which mirrors the encoder structure with eight circular nodes and dense interconnections. From the right side of the decoder, connections converge into a final narrow vertical column labeled “x subscript k plus 1”, containing three circular nodes inside a red-outlined box.

Architecture of the proposed DK-MPC framework. The encoder maps the quadrotor state xk into a latent space representation zk. The dynamics in this latent space are modeled linearly using matrices A(θA) and B(θB), which evolve the system according to the control input uk. The decoder then reconstructs the next state xk+1 from the predicted latent state zk+1. This integration of neural networks with Koopman-based linear dynamics enables accurate forward prediction and serves as the foundation for the DK-MPC controller

Figure 3
A neural network diagram with layered nodes, transformation blocks, and a summation node connected in sequence.On the far left, a vertical rectangular column labeled “x subscript k” contains three circular nodes arranged vertically inside a narrow red-outlined box. From this column, multiple diagonal lines fan out to the right into a larger rectangular block labeled “Encoder”, which is outlined in green. Inside the encoder, eight circular nodes are arranged vertically, with dense diagonal connections forming a layered neural network pattern. From the right side of the encoder, connections lead into a narrow vertical column of five circular nodes labeled “z subscript k”, enclosed in a light yellow box. From this column, a horizontal arrow extends to the right into a square block labeled “A of (theta subscript A)”, which contains a grid pattern of small squares. A line exits the right side of this block and curves downward toward a circular summation node marked with a plus symbol. Below the “z subscript k” column, a second narrow vertical column labeled “u subscript k” contains four circular nodes inside a blue-outlined box. A horizontal arrow extends from this column into another square block labeled “B of (theta subscript B)”, which also contains a grid pattern. A line exits this block and curves upward into the same circular summation node. From the summation node, a horizontal arrow extends to the right into another narrow vertical column labeled “z subscript k plus 1”, again shown as stacked five circular nodes inside a light yellow box. From this column, multiple diagonal connections spread into a second large green-outlined rectangular block labeled “Decoder”, which mirrors the encoder structure with eight circular nodes and dense interconnections. From the right side of the decoder, connections converge into a final narrow vertical column labeled “x subscript k plus 1”, containing three circular nodes inside a red-outlined box.

Architecture of the proposed DK-MPC framework. The encoder maps the quadrotor state xk into a latent space representation zk. The dynamics in this latent space are modeled linearly using matrices A(θA) and B(θB), which evolve the system according to the control input uk. The decoder then reconstructs the next state xk+1 from the predicted latent state zk+1. This integration of neural networks with Koopman-based linear dynamics enables accurate forward prediction and serves as the foundation for the DK-MPC controller

Close Figure 3
Figure 4
A graph with four decreasing curves over epochs showing different loss trends.The horizontal axis is labeled “Epoch” and ranges from 0 to 50 with an interval of 10. The vertical axis is labeled “M S E Loss” and ranges from 0 to 2 with an interval of 0.2. Four curves are plotted and identified in the legend as “Linear loss”, “Stability loss”, “Reconstruction loss”, and “Total loss”. The “Linear loss” and “Total loss” curves start at higher values near epoch 0, with the total loss beginning close to about 1.9 and the linear loss slightly lower near about 1.85. Both curves decrease sharply within the first few epochs and approach near zero by around epoch 5, then remain nearly flat along the horizontal axis for the rest of the range. The “Reconstruction loss” curve starts at a lower value near approximately 0.1 at epoch 0 and decreases quickly to near zero within the first few epochs, then stays close to zero. The “Stability loss” curve begins very close to zero at epoch 0 and remains near the horizontal axis throughout, with only a slight change visible near the initial epochs. All four curves are smooth and decreasing, with rapid decline occurring between epochs 0 and 5 and a nearly flat trend from approximately epoch 5 to 50. Note: All numerical data values are approximated.

Training loss convergence of the DK-MPC model over 50 epochs. The plot illustrates the mean squared error (MSE) for individual loss components – linear dynamics loss, stability loss and reconstruction loss – as well as the total loss. All components show rapid convergence within the first few epochs, indicating stable and efficient learning

Figure 4
A graph with four decreasing curves over epochs showing different loss trends.The horizontal axis is labeled “Epoch” and ranges from 0 to 50 with an interval of 10. The vertical axis is labeled “M S E Loss” and ranges from 0 to 2 with an interval of 0.2. Four curves are plotted and identified in the legend as “Linear loss”, “Stability loss”, “Reconstruction loss”, and “Total loss”. The “Linear loss” and “Total loss” curves start at higher values near epoch 0, with the total loss beginning close to about 1.9 and the linear loss slightly lower near about 1.85. Both curves decrease sharply within the first few epochs and approach near zero by around epoch 5, then remain nearly flat along the horizontal axis for the rest of the range. The “Reconstruction loss” curve starts at a lower value near approximately 0.1 at epoch 0 and decreases quickly to near zero within the first few epochs, then stays close to zero. The “Stability loss” curve begins very close to zero at epoch 0 and remains near the horizontal axis throughout, with only a slight change visible near the initial epochs. All four curves are smooth and decreasing, with rapid decline occurring between epochs 0 and 5 and a nearly flat trend from approximately epoch 5 to 50. Note: All numerical data values are approximated.

Training loss convergence of the DK-MPC model over 50 epochs. The plot illustrates the mean squared error (MSE) for individual loss components – linear dynamics loss, stability loss and reconstruction loss – as well as the total loss. All components show rapid convergence within the first few epochs, indicating stable and efficient learning

Close Figure 4
Figure 5
Four stacked graphs showing x, y, z, and phi versus sample index with two overlapping curves.Four vertically stacked graphs share a horizontal axis labeled “Sample Index” ranging from 0 to 14000 with an interval of 200. Each graph contains two curves labeled in the legend as “True” (dashed blue) and “Predicted” (solid orange), which closely overlap across most regions. In the top graph, the vertical axis is labeled “x (meters)” and ranges from negative 1 to 1 with an interval of 0.5. The curves oscillate with varying amplitude, showing frequent peaks and troughs between approximately negative 1 and 0.9 from sample index 0 to around 5000. Between 5000 and 9000, the oscillations become smaller and more irregular. From about 9000 to 11000, the curve rises gradually toward positive values near 0.5. After 11000, the curve drops sharply to around negative 1 near 12000 and then rises slightly toward the end. In the second graph, the vertical axis is labeled “y (meters)” and ranges from negative 1 to 1 with an interval of 0.5. The curves show oscillatory behavior with moderate amplitude. From 0 to around 4000, there are repeated peaks approaching 0.9 and troughs near negative 0.5. Between 4000 and 8000, the oscillations reduce in amplitude and become smoother. From 8000 to 12000, the curve shows smaller undulations around zero. Toward the end, the values fluctuate between approximately negative 0.5 and 0.3. In the third graph, the vertical axis is labeled “z (meters)” and ranges from negative 1 to 0 with an interval of 0.5. The curves remain mostly near negative 0.8 to negative 1 from 0 to around 9000, with small fluctuations. Around 10000, there is a sharp upward spike reaching close to 0, followed by rapid drops and rises between approximately negative 1 and negative 0.3. After 11000, the curve continues with irregular fluctuations and several peaks and dips before settling near negative values toward the end. In the bottom graph, the vertical axis is labeled “phi (radian)” and ranges from negative 0.2 to 0.4 with an interval of 0.2. From 0 to about 4000, the curves show oscillations between approximately negative 0.1 and positive 0.3. Between 4000 and 9000, the fluctuations become more irregular and dense, with frequent small peaks and dips across the range. After 9000, the curve stabilizes with smaller variations, remaining mostly between 0 and 0.1, with occasional minor deviations. Across all four graphs, the dashed blue and solid orange curves follow nearly identical paths with minimal visible separation. Note: All numerical data values are approximated.

Prediction performance of the trained DK-MPC model. The figure shows the comparison between true and predicted trajectories for position (x, y, z) and roll angle (ϕ) over time. The predicted signals (orange) closely follow the ground truth (blue dashed), demonstrating the model's ability to accurately capture the system dynamics in both translational and rotational dimensions

Figure 5
Four stacked graphs showing x, y, z, and phi versus sample index with two overlapping curves.Four vertically stacked graphs share a horizontal axis labeled “Sample Index” ranging from 0 to 14000 with an interval of 200. Each graph contains two curves labeled in the legend as “True” (dashed blue) and “Predicted” (solid orange), which closely overlap across most regions. In the top graph, the vertical axis is labeled “x (meters)” and ranges from negative 1 to 1 with an interval of 0.5. The curves oscillate with varying amplitude, showing frequent peaks and troughs between approximately negative 1 and 0.9 from sample index 0 to around 5000. Between 5000 and 9000, the oscillations become smaller and more irregular. From about 9000 to 11000, the curve rises gradually toward positive values near 0.5. After 11000, the curve drops sharply to around negative 1 near 12000 and then rises slightly toward the end. In the second graph, the vertical axis is labeled “y (meters)” and ranges from negative 1 to 1 with an interval of 0.5. The curves show oscillatory behavior with moderate amplitude. From 0 to around 4000, there are repeated peaks approaching 0.9 and troughs near negative 0.5. Between 4000 and 8000, the oscillations reduce in amplitude and become smoother. From 8000 to 12000, the curve shows smaller undulations around zero. Toward the end, the values fluctuate between approximately negative 0.5 and 0.3. In the third graph, the vertical axis is labeled “z (meters)” and ranges from negative 1 to 0 with an interval of 0.5. The curves remain mostly near negative 0.8 to negative 1 from 0 to around 9000, with small fluctuations. Around 10000, there is a sharp upward spike reaching close to 0, followed by rapid drops and rises between approximately negative 1 and negative 0.3. After 11000, the curve continues with irregular fluctuations and several peaks and dips before settling near negative values toward the end. In the bottom graph, the vertical axis is labeled “phi (radian)” and ranges from negative 0.2 to 0.4 with an interval of 0.2. From 0 to about 4000, the curves show oscillations between approximately negative 0.1 and positive 0.3. Between 4000 and 9000, the fluctuations become more irregular and dense, with frequent small peaks and dips across the range. After 9000, the curve stabilizes with smaller variations, remaining mostly between 0 and 0.1, with occasional minor deviations. Across all four graphs, the dashed blue and solid orange curves follow nearly identical paths with minimal visible separation. Note: All numerical data values are approximated.

Prediction performance of the trained DK-MPC model. The figure shows the comparison between true and predicted trajectories for position (x, y, z) and roll angle (ϕ) over time. The predicted signals (orange) closely follow the ground truth (blue dashed), demonstrating the model's ability to accurately capture the system dynamics in both translational and rotational dimensions

Close Figure 5
Figure 6
Four stacked graphs of x, y, z, and phi versus time with three overlapping curves.Four vertically stacked graphs share a horizontal axis labeled “Time (seconds)” ranging from 0 to 50 with an interval of 10. Each graph contains three curves identified in the legend as “Reference” (dashed blue), “Nonlinear M P C” (solid orange), and “D K M P C” (solid green). In the top graph, the vertical axis is labeled “x (meters)” and ranges from 0 to 3 with an interval of 1. The dashed blue reference and nonlinear M P C curves overlap and form step-like levels: near 0.4 from 0 to about 10 seconds, rising to around 0.7 from 10 to 20 seconds, then to about 0.8 from 20 to 30 seconds, and dropping back near 0.4 after 30 seconds. The orange curve shows a sharp peak above 2.3 near the start, then quickly settles and follows the step levels with small oscillations around each transition. The green curve follows the step pattern more closely with smaller overshoot at transitions. In the second graph, the vertical axis is labeled “y (meters)” and ranges from 0 to 1 with an interval of 0.5. The dashed blue reference and nonlinear M P C curves overlap and show step changes: around 0.4 from 0 to 10 seconds, rising to about 0.7 from 10 to 20 seconds, dropping to around 0.4 from 20 to 30 seconds, and increasing again to about 0.7 after 30 seconds. The orange curve exhibits oscillations near the start, including a peak above 1, then settles and tracks the step levels with mild overshoot at transitions. The green curve follows the same step pattern with smaller deviations and smoother transitions. In the third graph, the vertical axis is labeled “z (meters)” and ranges from 0 to 4 with an interval of 2. The dashed blue reference and nonlinear M P C curves overlap and remain near about 2.5 from 0 to 10 seconds, increase slightly to around 2.8 between 10 and 30 seconds, and then step up to about 3.2 after 30 seconds. The orange curve shows a large initial rise from near 0 to above 2, followed by oscillations before settling close to the reference levels. The green curve tracks the reference closely with minimal deviation and small adjustments at the step near 30 seconds. In the bottom graph, the vertical axis is labeled “phi (radian)” and ranges from 0 to 4 with an interval of 2. The dashed blue reference remains near 1 from 0 to 20 seconds, then drops to near 0 from 20 to 30 seconds, and stays close to 0 afterward. The orange curve shows initial oscillations near 0, then remains close to the reference with small fluctuations. The green curve shows sharp spikes at the beginning and around 30 seconds, reaching values above 4, and then quickly returns toward the reference level. Across all four graphs, the three curves are closely aligned after initial transients, with differences mainly visible near the beginning and at step transitions around 10, 20, and 30 seconds. Note: All numerical data values are approximated.

Point stabilization performance of the DK-MPC (green) compared to nonlinear MPC (orange). The plots show the evolution of position (x, y and z) and roll angle (ϕ) as the quadrotor tracks stepwise changes (blue dashed) in the target position and orientation

Figure 6
Four stacked graphs of x, y, z, and phi versus time with three overlapping curves.Four vertically stacked graphs share a horizontal axis labeled “Time (seconds)” ranging from 0 to 50 with an interval of 10. Each graph contains three curves identified in the legend as “Reference” (dashed blue), “Nonlinear M P C” (solid orange), and “D K M P C” (solid green). In the top graph, the vertical axis is labeled “x (meters)” and ranges from 0 to 3 with an interval of 1. The dashed blue reference and nonlinear M P C curves overlap and form step-like levels: near 0.4 from 0 to about 10 seconds, rising to around 0.7 from 10 to 20 seconds, then to about 0.8 from 20 to 30 seconds, and dropping back near 0.4 after 30 seconds. The orange curve shows a sharp peak above 2.3 near the start, then quickly settles and follows the step levels with small oscillations around each transition. The green curve follows the step pattern more closely with smaller overshoot at transitions. In the second graph, the vertical axis is labeled “y (meters)” and ranges from 0 to 1 with an interval of 0.5. The dashed blue reference and nonlinear M P C curves overlap and show step changes: around 0.4 from 0 to 10 seconds, rising to about 0.7 from 10 to 20 seconds, dropping to around 0.4 from 20 to 30 seconds, and increasing again to about 0.7 after 30 seconds. The orange curve exhibits oscillations near the start, including a peak above 1, then settles and tracks the step levels with mild overshoot at transitions. The green curve follows the same step pattern with smaller deviations and smoother transitions. In the third graph, the vertical axis is labeled “z (meters)” and ranges from 0 to 4 with an interval of 2. The dashed blue reference and nonlinear M P C curves overlap and remain near about 2.5 from 0 to 10 seconds, increase slightly to around 2.8 between 10 and 30 seconds, and then step up to about 3.2 after 30 seconds. The orange curve shows a large initial rise from near 0 to above 2, followed by oscillations before settling close to the reference levels. The green curve tracks the reference closely with minimal deviation and small adjustments at the step near 30 seconds. In the bottom graph, the vertical axis is labeled “phi (radian)” and ranges from 0 to 4 with an interval of 2. The dashed blue reference remains near 1 from 0 to 20 seconds, then drops to near 0 from 20 to 30 seconds, and stays close to 0 afterward. The orange curve shows initial oscillations near 0, then remains close to the reference with small fluctuations. The green curve shows sharp spikes at the beginning and around 30 seconds, reaching values above 4, and then quickly returns toward the reference level. Across all four graphs, the three curves are closely aligned after initial transients, with differences mainly visible near the beginning and at step transitions around 10, 20, and 30 seconds. Note: All numerical data values are approximated.

Point stabilization performance of the DK-MPC (green) compared to nonlinear MPC (orange). The plots show the evolution of position (x, y and z) and roll angle (ϕ) as the quadrotor tracks stepwise changes (blue dashed) in the target position and orientation

Close Figure 6
Figure 7
A complex plane plot with scattered eigenvalue points inside a circular boundary.The horizontal axis is labeled “Real Part” and ranges from negative 1.0 to 1.0 with an interval of 0.5. The vertical axis is labeled “Imaginary Part” and ranges from negative 1.0 to 1.0 with an interval of 0.5. A dashed circular boundary is centered at the origin with radius 1, forming a unit circle. Solid horizontal and vertical lines intersect at the origin, dividing the plane into four quadrants. Multiple red cross markers labeled “Eigenvalues (lambda)” are scattered within the circle. Several points are clustered near the right side close to the horizontal axis around a real value slightly greater than 0.6 and an imaginary value near 0. Additional points appear distributed across all quadrants, including positions near negative real values around negative 0.6 with both positive and negative imaginary parts, as well as points near the vertical axis slightly above and below the origin from between negative 0.5 and 0.5. The distribution forms a loose, uneven pattern with a noticeable concentration toward the right side. All points lie within the dashed unit circle labeled “Unit Circle (Boundary)”. Note: All numerical data values are approximated.

Spectral analysis of the learned Koopman transition matrix A. The red markers denote the eigenvalues (λi) in the complex plane

Figure 7
A complex plane plot with scattered eigenvalue points inside a circular boundary.The horizontal axis is labeled “Real Part” and ranges from negative 1.0 to 1.0 with an interval of 0.5. The vertical axis is labeled “Imaginary Part” and ranges from negative 1.0 to 1.0 with an interval of 0.5. A dashed circular boundary is centered at the origin with radius 1, forming a unit circle. Solid horizontal and vertical lines intersect at the origin, dividing the plane into four quadrants. Multiple red cross markers labeled “Eigenvalues (lambda)” are scattered within the circle. Several points are clustered near the right side close to the horizontal axis around a real value slightly greater than 0.6 and an imaginary value near 0. Additional points appear distributed across all quadrants, including positions near negative real values around negative 0.6 with both positive and negative imaginary parts, as well as points near the vertical axis slightly above and below the origin from between negative 0.5 and 0.5. The distribution forms a loose, uneven pattern with a noticeable concentration toward the right side. All points lie within the dashed unit circle labeled “Unit Circle (Boundary)”. Note: All numerical data values are approximated.

Spectral analysis of the learned Koopman transition matrix A. The red markers denote the eigenvalues (λi) in the complex plane

Close Figure 7
Figure 8
Four time-series plots showing x, y, z, and phi trajectories with three curves.Four vertically stacked graphs are shown, each with a horizontal axis labeled “Time (seconds)” ranging from 0 to 50 with an interval of 10. A legend in the top graph identifies three curves: “Reference” shown as a dashed blue line, “Nonlinear M P C” shown as a solid orange line, and “D K M P C” shown as a solid green line. In the top graph, the vertical axis is labeled “x (meters)” and ranges from 0 to 3 with an interval of 1. The dashed blue curve begins near 0.5, increases gradually toward about 0.9 around the middle, and then slightly declines toward about 0.4 by the end. The green curve rises smoothly and closely follows the dashed blue curve. The orange curve shows a sharp spike above 2.3 near the start, then settles and follows the same general path with small oscillations. In the second graph, the vertical axis is labeled “y (meters)” and ranges from 0 to 1.5 with an interval of 0.5. The dashed blue curve rises from near 0 to about 0.8 around 20 seconds, then drops toward about 0.3 near 30 seconds, and rises again toward about 0.7. The green curve follows the same overall pattern as the dashed blue curve with smoother transitions. The orange curve shows an early peak above 1.1 and then fluctuates around the dashed blue curve with visible oscillations. In the third graph, the vertical axis is labeled “z (meters)” and ranges from 0 to 4 with an interval of 1. The dashed blue curve increases gradually from about 2.1 to around 3.2 over time. The green curve closely follows the dashed blue curve with minimal variation. The orange curve rises sharply from near 0 to above 2.7 early on, then aligns with the gradual upward trend. In the bottom graph, the vertical axis is labeled “phi (radian)” and ranges from negative 2 to 4 with an interval of 2. The dashed blue curve forms a repeating oscillatory pattern with multiple peaks above 2 and dips below 0. The orange curve remains near 0 with small fluctuations in the early stage. The green curve follows the oscillatory pattern closely, matching the peaks and troughs in position and shape. All four graphs show three overlapping curves with distinct behaviors at the beginning and smoother alignment over the remaining time range. Note: All numerical data values are approximated.

Comparison of control performance between the proposed DK-MPC (green) and a standard nonlinear MPC (orange) in a trajectory tracking task (blue dashed)

Figure 8
Four time-series plots showing x, y, z, and phi trajectories with three curves.Four vertically stacked graphs are shown, each with a horizontal axis labeled “Time (seconds)” ranging from 0 to 50 with an interval of 10. A legend in the top graph identifies three curves: “Reference” shown as a dashed blue line, “Nonlinear M P C” shown as a solid orange line, and “D K M P C” shown as a solid green line. In the top graph, the vertical axis is labeled “x (meters)” and ranges from 0 to 3 with an interval of 1. The dashed blue curve begins near 0.5, increases gradually toward about 0.9 around the middle, and then slightly declines toward about 0.4 by the end. The green curve rises smoothly and closely follows the dashed blue curve. The orange curve shows a sharp spike above 2.3 near the start, then settles and follows the same general path with small oscillations. In the second graph, the vertical axis is labeled “y (meters)” and ranges from 0 to 1.5 with an interval of 0.5. The dashed blue curve rises from near 0 to about 0.8 around 20 seconds, then drops toward about 0.3 near 30 seconds, and rises again toward about 0.7. The green curve follows the same overall pattern as the dashed blue curve with smoother transitions. The orange curve shows an early peak above 1.1 and then fluctuates around the dashed blue curve with visible oscillations. In the third graph, the vertical axis is labeled “z (meters)” and ranges from 0 to 4 with an interval of 1. The dashed blue curve increases gradually from about 2.1 to around 3.2 over time. The green curve closely follows the dashed blue curve with minimal variation. The orange curve rises sharply from near 0 to above 2.7 early on, then aligns with the gradual upward trend. In the bottom graph, the vertical axis is labeled “phi (radian)” and ranges from negative 2 to 4 with an interval of 2. The dashed blue curve forms a repeating oscillatory pattern with multiple peaks above 2 and dips below 0. The orange curve remains near 0 with small fluctuations in the early stage. The green curve follows the oscillatory pattern closely, matching the peaks and troughs in position and shape. All four graphs show three overlapping curves with distinct behaviors at the beginning and smoother alignment over the remaining time range. Note: All numerical data values are approximated.

Comparison of control performance between the proposed DK-MPC (green) and a standard nonlinear MPC (orange) in a trajectory tracking task (blue dashed)

Close Figure 8
Figure 9
A dual-axis line graph comparing R-squared scores and computation time across prediction horizons.The horizontal axis is labeled “Prediction Horizon (H)” and ranges from 5 to 25 with an interval of 5. The left vertical axis is labeled “R-squared Score” and ranges from negative 2.5 to 2 with an interval of 0.5, and the right vertical axis is labeled “Computation Time per Step (milliseconds)” and ranges from 0 to 60 with an interval of 10. Four curves are plotted and identified in the legend. The “Nonlinear M P C R-squared” curve is shown as a solid orange line with square markers. It appears only at higher horizons, starting from H equals 20 at negative 2.5 of score and rising vertically to around negative 0.2 and then sloping downward toward the right at H equals 25 around negative 0.5 score. The “D K-M P C R-squared” curve is shown as a solid green line with circular markers. It remains constant at a value of 1 across all horizons from 5 through 25, forming a horizontal line. The “Nonlinear M P C Time” curve is shown as a dashed blue line with square markers. It starts near about 15 milliseconds at H equals 5, increases to around 20 milliseconds at H equals 10, rises further to about 39 milliseconds at H equals 15, peaks near 58 milliseconds at H equals 20, and then decreases to about 48 milliseconds at H equals 25, forming a rising, then slightly falling, trend. The “D K-M P C Time” curve is shown as a dashed purple line with circular markers. It remains low across all horizons, starting near about 2 milliseconds at H equals 5 and gradually increasing slightly to around 3 milliseconds by H equals 25. The curves are clearly separated by color and marker style, with R-squared values referenced to the left axis and computation time values referenced to the right axis. Note: All numerical data values are approximated.

Comparison of DK-MPC and nonlinear MPC performance across varying prediction horizon lengths (H = 5 to 25). The left y-axis shows the R2 score, indicating tracking accuracy, while the right y-axis shows the computation time per control step in milliseconds. DK-MPC maintains consistently high R2 scores and low computation time across all horizons, while Nonlinear MPC exhibits significant instability in both accuracy and runtime. In the nonlinear MPC approach where H < 20, the R2 error is significantly negative, implying that predictions are very far off from the actual values, failing to find solutions

Figure 9
A dual-axis line graph comparing R-squared scores and computation time across prediction horizons.The horizontal axis is labeled “Prediction Horizon (H)” and ranges from 5 to 25 with an interval of 5. The left vertical axis is labeled “R-squared Score” and ranges from negative 2.5 to 2 with an interval of 0.5, and the right vertical axis is labeled “Computation Time per Step (milliseconds)” and ranges from 0 to 60 with an interval of 10. Four curves are plotted and identified in the legend. The “Nonlinear M P C R-squared” curve is shown as a solid orange line with square markers. It appears only at higher horizons, starting from H equals 20 at negative 2.5 of score and rising vertically to around negative 0.2 and then sloping downward toward the right at H equals 25 around negative 0.5 score. The “D K-M P C R-squared” curve is shown as a solid green line with circular markers. It remains constant at a value of 1 across all horizons from 5 through 25, forming a horizontal line. The “Nonlinear M P C Time” curve is shown as a dashed blue line with square markers. It starts near about 15 milliseconds at H equals 5, increases to around 20 milliseconds at H equals 10, rises further to about 39 milliseconds at H equals 15, peaks near 58 milliseconds at H equals 20, and then decreases to about 48 milliseconds at H equals 25, forming a rising, then slightly falling, trend. The “D K-M P C Time” curve is shown as a dashed purple line with circular markers. It remains low across all horizons, starting near about 2 milliseconds at H equals 5 and gradually increasing slightly to around 3 milliseconds by H equals 25. The curves are clearly separated by color and marker style, with R-squared values referenced to the left axis and computation time values referenced to the right axis. Note: All numerical data values are approximated.

Comparison of DK-MPC and nonlinear MPC performance across varying prediction horizon lengths (H = 5 to 25). The left y-axis shows the R2 score, indicating tracking accuracy, while the right y-axis shows the computation time per control step in milliseconds. DK-MPC maintains consistently high R2 scores and low computation time across all horizons, while Nonlinear MPC exhibits significant instability in both accuracy and runtime. In the nonlinear MPC approach where H < 20, the R2 error is significantly negative, implying that predictions are very far off from the actual values, failing to find solutions

Close Figure 9
Figure 10
Four 3 D trajectory views with corresponding x, y, z, and phi plots below.The composite layout is arranged in two rows. The top row contains four three-dimensional trajectory plots labeled “(a)”, “(b)”, “(c)”, and “(d)”. Each 3 D plot has the same axes: the bottom left axis is labeled “X (meters)” and ranges from negative 5 to 5 with an interval of 5; the bottom right axis is labeled “Y (meters)” and ranges from negative 5 to 0 with an interval of 5; and the vertical axis on the left is labeled “Z (meters)” and ranges from 0 to 12 with an interval of 2. In all 3 D plots, a dashed red curve forms a smooth upward helical spiral, looping around a central axis while rising vertically. A solid blue curve follows a similar path. In panel (a), the blue trajectory starts below the spiral and slightly offset outward, then curves inward and upward toward the lower turns of the spiral. The quadrotor is positioned near the lower section, below the main spiral loop. In panel (b), the blue curve bends smoothly and attaches to the spiral path along a mid-level segment, following the curvature of one loop. The quadrotor is positioned along this mid-height section on the left where the blue and red curves nearly coincide. In panel (c), both curves trace a higher portion of the spiral, following a curved arc that wraps around the central axis with minimal separation. The quadrotor is located along this upper-middle segment, aligned with the direction of the spiral. In panel (d), the blue curve closely overlaps the red spiral along the upper loops, continuing the circular upward pattern. The quadrotor is positioned near the top of the spiral, where the path flattens slightly before continuing upward. Across all four panels, the blue trajectory shows a progression from an initial offset path to full alignment with the red helical spiral, maintaining the same circular looping pattern around the vertical axis. The bottom row contains four two-dimensional plots aligned with the panels above and labeled collectively as “(e)”. Each plot has a horizontal axis labeled “Sample” ranging from 0 to 1000 with an interval of 500. Each plot includes two curves mentioned in a legend identifying the dashed red curve as “Reference” and the solid blue curve as “D K-M P C”. In the first plot, the vertical axis is labeled “x (meters)” and ranges from negative 5 to 10 with an interval of 5. The dashed red and solid blue curves follow an oscillatory pattern, starting near about 5 at sample 0, decreasing to around negative 4 near sample 400, rising to about 5 near sample 800, and ending near 0. The blue curve closely follows the red curve with small deviations near the start. In the second plot, the vertical axis is labeled “y (meters)” and ranges from negative 4 to 4 with an interval of 2. The curves form a sinusoidal pattern, starting near negative 2, rising to about 3 near sample 200, dropping to around negative 4 near sample 600, and increasing again to about 3 near sample 1000. The blue curve closely tracks the red dashed curve and lies slightly above at the peaks. In the third plot, the vertical axis is labeled “z (meters)” and ranges from 0 to 10 with an interval of 2. Both curves show a steady increasing trend, starting near about 1 at sample 0 and rising to around 10 at sample 1000, with the blue curve almost completely overlapping the red curve. In the fourth plot, the vertical axis is labeled “phi (radian)” and ranges from negative 0.2 to 0.2 with an interval of 0.1. The curves exhibit periodic oscillations, starting near 0, rising to about 0.25, dropping to around negative 0.2 near sample 500, and rising again toward about 0.25 by sample 1000. The blue curve closely follows the red dashed curve throughout. Note: All numerical data values are approximated.

Helical trajectory following of the quadrotor using the proposed DK-MPC framework with prediction horizon H = 50. The red dashed line indicates the reference trajectory, while the blue line represents the actual quadrotor path. Snapshots (a)–(d) illustrate different stages of the helical maneuver. The controller achieves accurate real-time tracking with an average computation cost of 15 ms per control step. Plots in (e) shows the corresponding trajectory tracking in x, y, z and/phi

Figure 10
Four 3 D trajectory views with corresponding x, y, z, and phi plots below.The composite layout is arranged in two rows. The top row contains four three-dimensional trajectory plots labeled “(a)”, “(b)”, “(c)”, and “(d)”. Each 3 D plot has the same axes: the bottom left axis is labeled “X (meters)” and ranges from negative 5 to 5 with an interval of 5; the bottom right axis is labeled “Y (meters)” and ranges from negative 5 to 0 with an interval of 5; and the vertical axis on the left is labeled “Z (meters)” and ranges from 0 to 12 with an interval of 2. In all 3 D plots, a dashed red curve forms a smooth upward helical spiral, looping around a central axis while rising vertically. A solid blue curve follows a similar path. In panel (a), the blue trajectory starts below the spiral and slightly offset outward, then curves inward and upward toward the lower turns of the spiral. The quadrotor is positioned near the lower section, below the main spiral loop. In panel (b), the blue curve bends smoothly and attaches to the spiral path along a mid-level segment, following the curvature of one loop. The quadrotor is positioned along this mid-height section on the left where the blue and red curves nearly coincide. In panel (c), both curves trace a higher portion of the spiral, following a curved arc that wraps around the central axis with minimal separation. The quadrotor is located along this upper-middle segment, aligned with the direction of the spiral. In panel (d), the blue curve closely overlaps the red spiral along the upper loops, continuing the circular upward pattern. The quadrotor is positioned near the top of the spiral, where the path flattens slightly before continuing upward. Across all four panels, the blue trajectory shows a progression from an initial offset path to full alignment with the red helical spiral, maintaining the same circular looping pattern around the vertical axis. The bottom row contains four two-dimensional plots aligned with the panels above and labeled collectively as “(e)”. Each plot has a horizontal axis labeled “Sample” ranging from 0 to 1000 with an interval of 500. Each plot includes two curves mentioned in a legend identifying the dashed red curve as “Reference” and the solid blue curve as “D K-M P C”. In the first plot, the vertical axis is labeled “x (meters)” and ranges from negative 5 to 10 with an interval of 5. The dashed red and solid blue curves follow an oscillatory pattern, starting near about 5 at sample 0, decreasing to around negative 4 near sample 400, rising to about 5 near sample 800, and ending near 0. The blue curve closely follows the red curve with small deviations near the start. In the second plot, the vertical axis is labeled “y (meters)” and ranges from negative 4 to 4 with an interval of 2. The curves form a sinusoidal pattern, starting near negative 2, rising to about 3 near sample 200, dropping to around negative 4 near sample 600, and increasing again to about 3 near sample 1000. The blue curve closely tracks the red dashed curve and lies slightly above at the peaks. In the third plot, the vertical axis is labeled “z (meters)” and ranges from 0 to 10 with an interval of 2. Both curves show a steady increasing trend, starting near about 1 at sample 0 and rising to around 10 at sample 1000, with the blue curve almost completely overlapping the red curve. In the fourth plot, the vertical axis is labeled “phi (radian)” and ranges from negative 0.2 to 0.2 with an interval of 0.1. The curves exhibit periodic oscillations, starting near 0, rising to about 0.25, dropping to around negative 0.2 near sample 500, and rising again toward about 0.25 by sample 1000. The blue curve closely follows the red dashed curve throughout. Note: All numerical data values are approximated.

Helical trajectory following of the quadrotor using the proposed DK-MPC framework with prediction horizon H = 50. The red dashed line indicates the reference trajectory, while the blue line represents the actual quadrotor path. Snapshots (a)–(d) illustrate different stages of the helical maneuver. The controller achieves accurate real-time tracking with an average computation cost of 15 ms per control step. Plots in (e) shows the corresponding trajectory tracking in x, y, z and/phi

Close Figure 10
Table 1

Hyperparameters for DK-MPC model training

HyperparameterValue
Learning rate1 × 10−4
Batch size32
Latent dimension8
Training epochs50
OptimizerAdam
Loss coefficient λ1, λ2, λ31, 50, 1
Regularization coefficient λ41 × 10−4
Table 2

Sensitivity analysis of the Koopman latent space dimension on prediction accuracy and computation time

DimensionAverage R2 scoreComputation time (ms)
471.32%17
686.13%7.9
899.45%8.0
1299.57%12.7
1699.57%19.7

Supplements

Supplementary data

References

1.
Foehn
 
P
,
Kaufmann
 
E
,
Romero
 
A
,
Penicka
 
R
,
Sun
 
S
,
Bauersfeld
 
L
,
Laengle
 
T
,
Cioffi
 
G
,
Song
 
Y
,
Loquercio
 
A
,
Scaramuzza
 
D
.
Agilicious: open-source and open-hardware agile quadrotor for vision-based flight
.
Sci Robot
.
2022
;
7
(
67
): eabl6259. doi: .
2.
Zulu
 
A
,
John
 
S
.
A review of control algorithms for autonomous quadrotors
.
Open J Appl Sci
.
2014
;
4
(
14
):
547
-
56
. doi: .
3.
Aizelman
 
I
,
Magazinnik
 
D
,
Feldman
 
D
,
Klein
 
I
.
Quadrotor with wheels: design and experimental evaluation
.
Sci Rep
.
2024
;
14
(
1
): 15603. doi: .
4.
Saunders
 
J
,
Saeedi
 
S
,
Li
 
W
.
Autonomous aerial robotics for package delivery: a technical review
.
J Field Robot
.
2024
;
41
(
1
):
3
-
49
. doi: .
5.
Kokate
 
P
,
Middey
 
A
,
Sadistap
 
S
,
Sarode
 
G
,
Narayan
 
A
.
Review on drone-assisted air-quality monitoring systems
.
Drones Autonomous Vehicles
.
2023
;
1
(
1
): 10005. doi: .
6.
Lyu
 
M
,
Zhao
 
Y
,
Huang
 
C
,
Huang
 
H
.
Unmanned aerial vehicles for search and rescue: a survey
.
Remote Sens
.
2023
;
15
(
13
):
3266
. doi: .
7.
Khalid
 
A
,
Mushtaq
 
Z
,
Arif
 
S
,
Zeb
 
K
,
Khan
 
MA
,
Bakshi
 
S
.
Control schemes for quadrotor UAV: taxonomy and survey
.
ACM Comput Surv
.
2023
;
56
(
5
):
1
-
32
. doi: .
8.
Dhadekar
 
DD
,
Sanghani
 
PD
,
Mangrulkar
 
K
,
Talole
 
S
.
Robust control of quadrotor using uncertainty and disturbance estimation
.
J Intell Robot Syst
.
2021
;
101
(
3
):
60
. doi: .
9.
Hanover
 
D
,
Foehn
 
P
,
Sun
 
S
,
Kaufmann
 
E
,
Scaramuzza
 
D
.
Performance, precision, and payloads: adaptive nonlinear MPC for quadrotors
.
IEEE Robot Automation Lett
.
2021
;
7
(
2
):
690
-
7
. doi: .
10.
Wang
 
C
,
Song
 
B
,
Huang
 
P
,
Tang
 
C
.
Trajectory tracking control for quadrotor robot subject to payload variation and wind gust disturbance
.
J Intell Robot Syst
.
2016
;
83
(
2
):
315
-
33
. doi: .
11.
Sönmez
 
S
,
Rutherford
 
MJ
,
Valavanis
 
KP
.
A survey of offline-and online-learning-based algorithms for multirotor uavs
.
Drones
.
2024
;
8
(
4
):
116
. doi: .
12.
Belkheiri
 
M
,
Rabhi
 
A
,
El Hajjaji
 
A
,
Pegard
 
C
.
Different linearization control techniques for a quadrotor system
. In:
CCCA12
.
IEEE
;
2012
. p.
1
-
6
.
13.
Salzmann
 
T
,
Kaufmann
 
E
,
Arrizabalaga
 
J
,
Pavone
 
M
,
Scaramuzza
 
D
,
Ryll
 
M
.
Real-time neural MPC: deep learning model predictive control for quadrotors and agile robotic platforms
.
IEEE Robot Automation Lett
.
2023
;
8
(
4
):
2397
-
404
. doi: .
14.
Saviolo
 
A
,
Li
 
G
,
Loianno
 
G
.
Physics-inspired temporal learning of quadrotor dynamics for accurate model predictive trajectory tracking
. ,
2022
.
15.
Zhou
 
S
,
Helwa
 
MK
,
Schoellig
 
AP
.
An inversion-based learning approach for improving control in quadrotor systems
.
IEEE Robot Automation Lett
.
2018
;
3
(
3
):
2366
-
73
.
16.
El-Hussieny
 
H
.
Real-time deep learning-based model predictive control of a 3-dof biped robot leg
.
Sci Rep
.
2024
;
14
(
1
): 16243. doi: .
17.
El-Hussieny
 
H
,
Hameed
 
IA
.
Obstacle-aware navigation of soft growing robots via deep reinforcement learning
.
IEEE Access
.
2024
;
12
:
38192
-
201
. doi: .
18.
Hwangbo
 
J
,
Sa
 
I
,
Siegwart
 
R
,
Hutter
 
M
.
Control of a quadrotor with reinforcement learning
.
IEEE Robot Automation Lett
.
2017
;
2
(
4
):
2096
-
103
. doi: .
19.
Dionigi
 
A
,
Sun
 
H
,
Omerdic
 
E
,
Floreano
 
D
,
Carrillo
 
L
.
The power of input: benchmarking zero-shot sim-to-real transfer of reinforcement learning control policies for quadrotor control
. ,
2024
.
20.
Narayanan
 
SSKS.
,
Tellez-Castro
 
D
,
Sutavani
 
S
,
Vaidya
 
U
.
SE(3) Koopman-MPC: data-driven learning and control of quadrotor UAVs
.
IFAC-PapersOnLine
.
2023
;
56
(
3
):
607
-
12
. doi: .
21.
Abido
 
MA
,
Manaa
 
ZM
,
Abdallah
 
AM
,
Ali
 
SSA
.
Koopman-LQR controller for quadrotor UAVs from data
. ,
2024
.
22.
Martini
 
S
.
Koopman-based modeling for nonlinear control of multirotor UAVs
.
PhD thesis, University of Denver
,
2024
.
23.
Rajkumar
 
SM
,
Cheng
 
S
,
Hovakimyan
 
N
,
Goswami
 
D
.
Linear model predictive control for quadrotors with an analytically derived Koopman model
. ,
2024
.
24.
Shi
 
L
,
Haseli
 
M
,
Mamakoukas
 
G
,
Bruder
 
D
,
Abraham
 
I
,
Murphey
 
T
,
Cortés
 
J
,
Karydis
 
K
.
Koopman operators in robot learning: theory, algorithms, and applications
. ,
2024
.
25.
Li
 
Y
,
Zhu
 
Q
,
Elahi
 
A
.
Quadcopter trajectory tracking control based on flatness model predictive control and neural network
.
Actuators
.
2024
;
13
(
4
):
154
. doi: .
26.
Bruder
 
D
,
Fu
 
X
,
Gillespie
 
RB
,
Remy
 
CD
,
Vasudevan
 
R
.
Data-driven control of soft robots using koopman operator theory
.
IEEE Trans Robot
.
2020
;
37
(
3
):
948
-
61
. doi: .
27.
Shi
 
H
,
Meng
 
MQ-H
.
Deep koopman operator with control for nonlinear systems
.
IEEE Robot Automation Lett
.
2022
;
7
(
3
):
7700
-
7
. doi: .
28.
Ng
 
AY
.
Feature selection, l 1 vs. l 2 regularization, and rotational invariance
. In:
Proceedings of the Twenty-First International Conference on Machine Learning
;
2004
. p.
78
.
29.
Mohajerin
 
N
,
Mozifian
 
M
,
Waslander
 
S
.
Deep learning a quadrotor dynamic model for multi-step prediction
. In:
2018 IEEE International Conference on Robotics and Automation (ICRA)
.
IEEE
;
2018
. p.
2454
-
9
.
30.
Mohajerin
 
N
,
Waslander
 
SL
.
Multistep prediction of dynamic systems with recurrent neural networks
.
IEEE Trans Neural Netw Learn Syst
.
2019
;
30
(
11
):
3370
-
83
. doi: .
31.
Kumar
 
M
,
Stoll
 
R
,
Stoll
 
N
.
A min-max approach to fuzzy clustering, estimation, and identification
.
IEEE Trans Fuzzy Syst
.
2006
;
14
(
2
):
248
-
62
. doi: .
32.
Abdelaziz
 
H
,
Ahmed
 
A
,
El-Hussieny
 
H
.
Approximate neural network-based nonlinear model predictive control of soft continuum robots
. In:
2024 20th IEEE/ASME International Conference on Mechatronic and Embedded Systems and Applications (MESA)
.
IEEE
;
2024
. p.
1
-
7
.
33.
Li
 
S
,
Chen
 
C
,
Zheng
 
H
,
Liu
 
Y
,
Xu
 
Q
,
Li
 
K
.
Nonlinear data-driven predictive control for mixed platoons based on Koopman operator
. In:
Proceedings of the 8th CAA International Conference on Vehicular Control and Intelligence (CVCI)
.
IEEE
;
2024
. p.
1
-
6
.
34.
Li
 
S
,
Wang
 
J
,
Yang
 
K
,
Xu
 
Q
,
Wang
 
J
,
Li
 
K
.
Robust nonlinear data-driven predictive control for mixed vehicle platoons via Koopman operator and reachability analysis
. ,
2025
.
35
Lyu
 
H
,
Guo
 
Y
,
Liu
 
P
,
Zheng
 
N
,
Wang
 
T
,
Yue
 
Q
.
Mitigating traffic oscillations in mixed traffic flow with scalable deep Koopman predictive control
.
Adv Eng Inform
.
2026
;
71
:
104258
-
104279
.

Languages

or Create an Account

Close subscription notice
Close access options