File size: 2,616 Bytes
232be21
 
28d8c52
8ecc9b4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
---
license: apache-2.0
pipeline_tag: tabular-regression
datasets:
- Allanatrix/QST
tags:
- Physics
---
# NexaQST: Quantum State Tomography with Physics-Informed Neural Networks

**NexaQST** is a Quantum State Tomography model built using a Physics-Informed Neural Network (PINN) trained on synthetic 2-qubit experiments. This model leverages quantum mechanical priors such as positivity, Hermiticity, and trace constraints to ensure physically plausible reconstructions of quantum states.

---

## Model Overview

- **Task**: Reconstruct quantum states (density matrices) from tomographic measurement traces
- **System**: Simulated 2-qubit experiments
- **Architecture**: Physics-Informed Neural Network (PINN)
- **Constraints Embedded**:
  - Positivity
  - Hermiticity
  - Trace normalization (Tr(ฯ) = 1)

---

## Dataset Generation

- **Simulation Toolkit**: [`QuTiP`](https://qutip.org/)
- **Process**:
  1. Generated full 2-qubit tomography experiments via simulation
  2. Extracted and structured measurement traces
  3. Derived backward synthetic traces from known density matrices
  4. Created supervised training pairs: `(trace โ†’ density matrix)`
  5. Enforced physical constraints directly inside the model during training

---

## Model Input/Output

- **Input**: Vector of tomographic measurement traces
- **Output**: Reconstructed density matrix (complex-valued 4x4 for 2 qubits)

---

## Example Usage

```python
# Pseudocode for usage

from model import NexaQSTModel
import torch

model = NexaQSTModel()
model.load_state_dict(torch.load("nexaqst_model.pt"))
model.eval()

traces = load_measurement_vector("qst_trace.npy")  # shape: (N,)
rho_pred = model.predict_density_matrix(traces)

# rho_pred is a 4x4 complex-valued matrix satisfying physical constraints
````

---

## Applications

* Quantum error correction diagnostics
* Quantum system identification
* Educational simulation of QST techniques
* Physically consistent state estimation for quantum simulations

---

## Licensing & Citation

* If used in publications or products, please credit the Nexa Scientific Suite.

---

## ๐ŸŒ Related Tools

* [๐Ÿ“ Nexa Data Studio](https://huggingface.co/spaces/Allanatrix/NexaDataStudio)
* [๐Ÿ”ฌ Nexa R\&D Hub](https://huggingface.co/spaces/Allanatrix/NexaR&D)
* [๐Ÿง  Nexa MOE Models](https://huggingface.co/collections/Allanatrix/nexa-models)
* [๐ŸŒ Nexa Hub (Main Portal)](https://huggingface.co/spaces/Allanatrix/NexaHub)

---

*Created by [Allan](https://huggingface.co/Allanatrix), independent quantum systems architect and ML researcher. Part of the Nexa scientific computing ecosystem.*