Emily
commited on
Commit
·
cc77288
1
Parent(s):
f826c3b
Update default epochs to 30 for all datasets and architectures
Browse files- Increase epochs slider maximum from 20 to 50
- Set default epochs value to 30 in UI
- Update all preset configurations (High Privacy, Balanced, High Utility) to use 30 epochs
- Update setOptimalParameters function to use 30 epochs
- Ensures consistent training duration across all dataset/architecture combinations
- app/static/js/main.js +4 -4
- app/templates/index.html +4 -4
app/static/js/main.js
CHANGED
@@ -71,21 +71,21 @@ class DPSGDExplorer {
|
|
71 |
noiseMultiplier: 1.5,
|
72 |
batchSize: 256,
|
73 |
learningRate: 0.005,
|
74 |
-
epochs:
|
75 |
},
|
76 |
'balanced': {
|
77 |
clippingNorm: 1.0,
|
78 |
noiseMultiplier: 1.0,
|
79 |
batchSize: 128,
|
80 |
learningRate: 0.01,
|
81 |
-
epochs:
|
82 |
},
|
83 |
'high-utility': {
|
84 |
clippingNorm: 1.5,
|
85 |
noiseMultiplier: 0.5,
|
86 |
batchSize: 64,
|
87 |
learningRate: 0.02,
|
88 |
-
epochs:
|
89 |
}
|
90 |
};
|
91 |
|
@@ -847,7 +847,7 @@ function setOptimalParameters() {
|
|
847 |
document.getElementById('noise-multiplier').value = '1.0'; // Moderate noise for good privacy
|
848 |
document.getElementById('batch-size').value = '256'; // Large batches for DP-SGD stability
|
849 |
document.getElementById('learning-rate').value = '0.05'; // Balanced learning rate
|
850 |
-
document.getElementById('epochs').value = '
|
851 |
|
852 |
// Update displays
|
853 |
updateClippingNormDisplay();
|
|
|
71 |
noiseMultiplier: 1.5,
|
72 |
batchSize: 256,
|
73 |
learningRate: 0.005,
|
74 |
+
epochs: 30
|
75 |
},
|
76 |
'balanced': {
|
77 |
clippingNorm: 1.0,
|
78 |
noiseMultiplier: 1.0,
|
79 |
batchSize: 128,
|
80 |
learningRate: 0.01,
|
81 |
+
epochs: 30
|
82 |
},
|
83 |
'high-utility': {
|
84 |
clippingNorm: 1.5,
|
85 |
noiseMultiplier: 0.5,
|
86 |
batchSize: 64,
|
87 |
learningRate: 0.02,
|
88 |
+
epochs: 30
|
89 |
}
|
90 |
};
|
91 |
|
|
|
847 |
document.getElementById('noise-multiplier').value = '1.0'; // Moderate noise for good privacy
|
848 |
document.getElementById('batch-size').value = '256'; // Large batches for DP-SGD stability
|
849 |
document.getElementById('learning-rate').value = '0.05'; // Balanced learning rate
|
850 |
+
document.getElementById('epochs').value = '30'; // Sufficient epochs for convergence
|
851 |
|
852 |
// Update displays
|
853 |
updateClippingNormDisplay();
|
app/templates/index.html
CHANGED
@@ -141,11 +141,11 @@
|
|
141 |
<span class="tooltip-text">Number of complete passes through the dataset. More epochs improves learning but increases privacy budget consumption.</span>
|
142 |
</span>
|
143 |
</label>
|
144 |
-
<input type="range" id="epochs" class="parameter-slider" min="1" max="
|
145 |
<div class="slider-display">
|
146 |
<span>1</span>
|
147 |
-
<span id="epochs-value">
|
148 |
-
<span>
|
149 |
</div>
|
150 |
</div>
|
151 |
|
@@ -214,7 +214,7 @@
|
|
214 |
<div id="training-status" class="status-badge" style="display: none;">
|
215 |
<span class="pulse"></span>
|
216 |
<span style="font-weight: 500; color: #4caf50;">Training in progress</span>
|
217 |
-
<span style="margin-left: auto; font-weight: 500;">Epoch: <span id="current-epoch">1</span> / <span id="total-epochs">
|
218 |
</div>
|
219 |
</div>
|
220 |
|
|
|
141 |
<span class="tooltip-text">Number of complete passes through the dataset. More epochs improves learning but increases privacy budget consumption.</span>
|
142 |
</span>
|
143 |
</label>
|
144 |
+
<input type="range" id="epochs" class="parameter-slider" min="1" max="50" step="1" value="30">
|
145 |
<div class="slider-display">
|
146 |
<span>1</span>
|
147 |
+
<span id="epochs-value">30</span>
|
148 |
+
<span>50</span>
|
149 |
</div>
|
150 |
</div>
|
151 |
|
|
|
214 |
<div id="training-status" class="status-badge" style="display: none;">
|
215 |
<span class="pulse"></span>
|
216 |
<span style="font-weight: 500; color: #4caf50;">Training in progress</span>
|
217 |
+
<span style="margin-left: auto; font-weight: 500;">Epoch: <span id="current-epoch">1</span> / <span id="total-epochs">30</span></span>
|
218 |
</div>
|
219 |
</div>
|
220 |
|