KIT | KIT-Bibliothek | Impressum | Datenschutz

Trained LSTM Ensemble Models for "Real-Time Prediction of Thermal History and Hardness in Laser Powder Bed Fusion Using Deep Learning"

Schüßler, Philipp ORCID iD icon 1; Schulze, Volker 1; Dietrich, Stefan ORCID iD icon 1
1 Institut für Angewandte Materialien – Werkstoffkunde (IAM-WK), Karlsruher Institut für Technologie (KIT)

Abstract (englisch):

This archive contains three sets of trained LSTM ensemble models for surrogate-based thermal history prediction during laser powder bed fusion (PBF-LB/M) of 42CrMo4 steel. Each ensemble consists of five independently seeded models trained with a six-stage curriculum that incrementally expands the training data selection. The three ensembles differ only in LSTM architecture depth and width, enabling a systematic comparison of model complexity. The trained weights are consumed by the companion framework via its inference and testing entry points.

Full-text publication: <added later>
Code-Repository: https://doi.org/10.35097/dg39f4p0wxqdnfxy
Training-Validation-Testing Dataset: https://doi.org/10.35097/pmem1cb9gu1ck8xz


Zugehörige Institution(en) am KIT Institut für Angewandte Materialien – Werkstoffkunde (IAM-WK)
Publikationstyp Forschungsdaten
Publikationsdatum 15.04.2026
Erstellungsdatum 09.04.2026
Identifikator DOI: 10.35097/37da9d66y4t27q55
KITopen-ID: 1000192076
Lizenz Creative Commons Namensnennung – Nicht kommerziell – Keine Bearbeitungen 4.0 International
Projektinformation 516837935 (DFG, DFG EIN, DI 2052/13-1)
Schlagwörter Additive Manufacturing, Deep Learning, Surrogate model, Carbon Steel, AISI 4140, 42CrMo4, Thermal history, Hardness, Trained model,
Liesmich

Contains 3 .zip files with the trained model ensembles and the torch.nn.Module class:

  • 01Layers16Cells_Ensemble.zip
  • 02Layers32Cells_Ensemble.zip
  • 04Layers64Cells_Ensemble.zip
  • recurrent_neuralnetworks.py

Loading a checkpoint manually:

import torch
from recurrent_neuralnetworks import LSTMModelWithTeacherForcing
model = LSTMModelWithTeacherForcing(
    num_features=12,   # 11 inputs + 1 temperature feedback
    num_hidden=32,     # match the run (16 / 32 / 64)
    num_layers=2,      # match the run (1 / 2 / 4)
    num_labels=1,
)
state = torch.load(&quot;path/to/model_best_....pth&quot;, map_location=&quot;cpu&quot;)
model.load_state_dict(state)
model.eval()
Art der Forschungsdaten Dataset
Relationen in KITopen
KIT – Die Universität in der Helmholtz-Gemeinschaft
KITopen Landing Page