Back to Home TR

TFT-Flood: AI-Powered Flood Early Warning System

District-level daily flood forecasting for Turkey, powered by a Temporal Fusion Transformer

Open Live Demo
Flood Risk Map
SIU 2026 (IEEE) — Accepted
This work has been accepted for publication as "TFT-Flood for Nationwide Flood Prediction in Turkey: A Three-Branch Hybrid Deep Learning Architecture" at the IEEE Signal Processing and Communications Applications Conference (SIU 2026).

Project Summary

This project is an AI system that performs daily flood risk prediction for districts across Turkey. The research model, TFT-Flood, is a three-branch hybrid deep learning architecture inspired by the Temporal Fusion Transformer, operating over 8-day time windows. It is evaluated on a dataset of 6,372 sequences covering 973 districts under a strict leak-free 1-day forecast protocol, where the prediction day is masked.

The 8-seed ensemble achieves 0.909 AUC (F1 0.84, Precision 0.88, Recall 0.81), outperforming all six benchmark baselines. The live system runs automatically every day, fetching current meteorological and hydrological data, performing district-level risk analysis, and generating an interactive choropleth map.

Technologies Used

Category Technologies
Model Architecture Temporal Fusion Transformer (TFT), Bi-LSTM, GRN/VSN — ~234K parameters
AI / ML Keras 3 (PyTorch backend), AdamW + cosine LR
Data Processing Pandas, NumPy, Scikit-learn
Visualization Folium (Leaflet.js), Matplotlib
Data Sources TUCBS open data (DSI + AFAD flood events, 2000–2020), ERA5 + NASA POWER, DSI discharge archive, Open-Meteo API
Hardware RTX 3050 Ti (4 GB VRAM) — single-sequence inference < 5 ms

Key Features

System Architecture

TFT-Flood analyzes an 8-day time window for each district through three parallel branches:

  1. Branch A — Static Enrichment: Processes static features through four GRN blocks to generate context vectors.
  2. Branch B — Temporal Processing: A Variable Selection Network (VSN) + Bi-LSTM encodes the temporal dynamics.
  3. Branch C — Risk Quantification: Concatenates the last timestep with static context and feeds a GRN-MLP head to yield a flood probability and a 4-class risk score.

Input Features: 7 dynamic (total precipitation, rain, max/min temperature, wind speed, soil moisture, DSI discharge) + 8 derived dynamic (3/5-day cumulative rain, rain change rate, temp×rain, soil×rain, rain intensity, temperature range, discharge×rain) = 15 dynamic; plus 8 static (altitude, slope, distance to nearest lake, water surface ratio, CORINE land cover).

Baseline Comparison

Comparison against six baselines on v8_full (6,372 sequences, 973 districts) under the leak-free 1-day forecast protocol. TFT-Flood (8-seed ensemble) attains the highest AUC and F1:

Model AUC F1 Precision Recall
Logistic Regression0.7870.7500.7210.782
Random Forest0.9050.8420.9130.782
XGBoost0.8970.8420.9070.786
Pure Bi-LSTM0.8910.8330.8740.795
ST-GNN0.8960.8340.9010.776
Informer-lite0.9000.8340.9010.776
TFT-Flood (ensemble)0.9090.8440.8840.807

Results

This project demonstrates the power of artificial intelligence in disaster management and early warning systems.